diff --git a/README.md b/README.md index f1bb8c2..860cfe4 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(Дигитално Процесиранје на Слика) +### 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. Using super-gradients as a framework for PyTorch, for ease of use and faster prototyping.