vesselDetection

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.