search

ZJU-lishuang / yolov5_prune

yolov5 prune，Support V2, V3, V4 and V6 versions of yolov5
Apache License 2.0
556 stars 136 forks source link

如何把剪枝后的模型，送入yolov5中的test.py中进行指标计算呢 #100

Open dddmmmyyy1998 opened 2 years ago

dddmmmyyy1998 commented 2 years ago

图片 报错如上。

ZJU-lishuang commented 2 years ago

refer to https://github.com/ZJU-lishuang/yolov5_prune/blob/v6/slim_prune_yolov5s_8x.py#L42 for val