search

hailo-ai / hailo_model_zoo

The Hailo Model Zoo includes pre-trained models and a full building and evaluation environment
MIT License
325 stars 45 forks source link

How to train yolov5m_wo_spp #104

Open GerHelWan opened 5 months ago

GerHelWan commented 5 months ago

Hi

I would like to train or fine-tune the yolov5m_wo_spp architecture that I have seen in the object detection benchmark. The mAP and the FPS for batch 1 sound promising.

See also https://github.com/hailo-ai/hailo_model_zoo/blob/master/docs/public_models/HAILO8/HAILO8_object_detection.rst

Unfortunately given the link https://github.com/ultralytics/yolov5/releases/tag/v2.0 I don't know exactly how to train the yolov5 m wo spp network. I guess spp stands for spatial pyramid pooling (spp) and wo for without?

Is it possible for you to describe what needs to be done in order to train the yolov5 m wo spp?

tmyapple commented 5 months ago

Hi @GerHelWan In order to retrain the model i suggest you to follow our documentation found here The directory comes with a Dockerfile to build a docker container with all necessary packages and files, including our own yolov5 configuration file and our final pt file to start the training from it