ZJU-lishuang / yolov5_prune

yolov5 prune,Support V2, V3, V4 and V6 versions of yolov5
Apache License 2.0
560 stars 137 forks source link

how do u get yolov5s_v3.cfg? #3

Closed codylcs closed 3 years ago

codylcs commented 3 years ago

thanks for your works, it is wonderful,but May i know how do u get yolov5s.cfg from yaml and .pt? if i want prune yolov5x.pt,what should i do? forward your reply

ZJU-lishuang commented 3 years ago

you should convert yaml to cfg.I rebuild the network layer by layer.

codylcs commented 3 years ago

sorry, i thought u convert by third-part tools,anyway,thanks for your reply