This project detects starfish by implementing Yolov5, which can be found here. There is no official research paper about Yolov5, but many people use it in practice, and it shows a great result.
- Training: detect-crown-of-thorns-starfish-infer.ipynb
- Inference: detect-crown-of-thorns-starfish-infer.ipynb
I achieved a F2 score of 0.494 on the validation set after 20 epochs. The F2 score weights recall more heavily than the precision, which makes sense because we want to detect as many starfish as we can.
