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