Running YOLOv5 on HRSID dataset for maritime vessel detection
YOLOv5's architecture consists of three main parts:
Backbone: This is the main body of the network. For YOLOv5, the backbone is designed using the New CSP-Darknet53 structure, a modification of the Darknet architecture used in previous versions. Neck: This part connects the backbone and the head. In YOLOv5, SPPF and New CSP-PAN structures are utilized. Head: This part is responsible for generating the final output. YOLOv5 uses the YOLOv3 Head for this purpose.
Dependancies: pycocotools, numpy, skimage, pandas, matplotlib, torch, wandb, yaml,tensorboard
https://api.wandb.ai/links/charak/kbb8sy5v
