AlibabaResearch / efficientteacher

A Supervised and Semi-Supervised Object Detection Library for YOLO Series
GNU General Public License v3.0
832 stars 151 forks source link

How to run model converter with yolov8n? #91

Open zhifeis opened 1 year ago

zhifeis commented 1 year ago

I've tried to run convert_pt_to_efficient.py to get my yolov8n .pt file to convert to the efficient teacher format but got stuck because map_v5.txt doesn't match the architecture of yolov8n. Where can I find this version of map.txt? Also, in the .yaml file aside from the depth and width multiple values should I modify anything else in the Model section for yolov8n? Thanks!

songjunnan123 commented 7 months ago

backbone.stage1 model.0 backbone.stage2_1 model.1 backbone.stage2_2 model.2 backbone.stage3_1 model.3 backbone.stage3_2 model.4 backbone.stage4_1 model.5 backbone.stage4_2 model.6 backbone.stage5_1 model.7 backbone.stage5_2 model.8 backbone.sppf model.9 neck.upsample1 model.10 neck.concat model.11 neck.C1 model.12 neck.upsample2 model.13 neck.concat model.14 neck.C2 model.15 neck.conv3 model.16 neck.concat model.17 neck.C3 model.18 neck.conv4 model.19 neck.concat model.20 neck.C4 model.21 head model.22