Paper: ShipYOLO: An Enhanced Model for Ship Detection

DOI: https://doi.org/10.1155/2021/1060182

Xu Han, "ShipYOLO: An Enhanced Model for Ship Detection," Journal of Advanced Transportation, vol. 2021, Article ID 1060182, 11 pages, 2021. https://doi.org/10.1155/2021/1060182.

1. Installation related environment

git clone https://github.com/hx-0614/ShipYOLO.git
cd ShipYOLO
pip install -r requirements.txt

2. Train

Pre training model yolov4.weights.

python train.py --rect

3. Test

python inference.py    # change ".pth" to deploy ".pth"

python test.py

4. Detect

python detect.py

5. Citation

...

Acknowledgements

https://hub.fastgit.org/WongKinYiu/PyTorch_YOLOv4