Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
robmarkcole committed Sep 25, 2021
1 parent d507601 commit 73aa66a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -57,4 +57,4 @@ Finally the model has mistaken the dock for a boat, which is perhaps understanda
</p>

## Conclusions
With relatively little time (1-2 hours including training) and effort a model was trained that provides decent performance for ships in open water. However small ships, clustered ships, and false positives close to land are an issue. Future work will evaluate the balance of the dataset with regard to these poorly performing cases. Note also that YOLOv5s is the smallest available, and better metrics could be achieved with one of the [larger models](https://github.com/ultralytics/yolov5#pretrained-checkpoints). Experimentation with alternative augmentation strategies could be explored.
With relatively little time (1-2 hours including training) and effort a model was trained that provides decent performance for ships in open water. However small ships, clustered ships, and false positives close to land are an issue. Future work will evaluate the balance of the dataset with regard to these poorly performing cases. Note also that YOLOv5s is the smallest available, and better metrics could be achieved with one of the [larger models](https://github.com/ultralytics/yolov5#pretrained-checkpoints). Experimentation with alternative augmentation strategies could be explored, to help balance out some of the variations we observe due to: sea conditions (rough or calm), lighting (overcast or bright sunshine), ship density.

0 comments on commit 73aa66a

Please sign in to comment.