diff --git a/README.md b/README.md index 94996cd..8b2c026 100644 --- a/README.md +++ b/README.md @@ -1,2 +1,5 @@ # vesselDetection -Vessel/Ship detection using YOLO +Ship detection using Machine Learning methods for course Digital Processing of Image(Дигитално Процесиранје на Слика) + +This is a simple ship detection model, made using pre-trained YOLO using COCO weights. +Using super-gradients as a framework for PyTorch, for ease of use and faster prototyping.