Gumpest / YOLOv5-Multibackbone-Compression

YOLOv5 Series Multi-backbone(TPH-YOLOv5, Ghostnet, ShuffleNetv2, Mobilenetv3Small, EfficientNetLite, PP-LCNet, SwinTransformer YOLO), Module(CBAM, DCN), Pruning (EagleEye, Network Slimming), Quantization (MQBench) and Deployment (TensorRT, ncnn) Compression Tool Box.
991 stars 201 forks source link

is:issue is:open detect,大佬您好为什么运行detect.py报错 return flops * img_size[0] / stride * img_size[1] / stride UnboundLocalError: local variable 'flops' referenced before assignment #28

Open ahhhxxx opened 2 years ago

ahhhxxx commented 2 years ago

is:issue is:open detect,大佬您好为什么运行detect.py报错 return flops img_size[0] / stride img_size[1] / stride UnboundLocalError: local variable 'flops' referenced before assignment image

Gumpest commented 2 years ago

原因:(https://github.com/Gumpest/YOLOv5-Multibackbone-Compression/blob/main/utils/torch_utils.py#L233) 应该是 from thop import profile没有执行 解决办法是:安装thop包

ahhhxxx commented 2 years ago

原因:( https://github.com/Gumpest/YOLOv5-Multibackbone-Compression/blob/main/utils/torch_utils.py#L233 ) 应该是 from thop import profile 没有执行 解决办法是:安装thop包

大佬安装了也不行呀 大佬建交流群吗? image

hjq0523 commented 2 years ago

我遇到了相同的问题,请问你们问题解决了嘛?

ahhhxxx commented 2 years ago

这是来自QQ邮箱的假期自动回复邮件。您好,我最近正在休假中,无法亲自回复您的邮件。我将在假期结束后,尽快给您回复。

hjq0523 commented 2 years ago

我对比https://github.com/ultralytics/yolov5/blob/master/utils/torch_utils.py 里的215行处model_info函数,只要将报错的245行注释掉,就可以测试成功。

ahhhxxx commented 2 years ago

我对比了https://github.com/ultralytics/yolov5/blob/master/utils/torch_utils.py 里的215行处model_info函数,只要将报错的245行注释掉,就可以测试成功。

我没有解决,但是如果注释掉了 就不能打印模型信息了

Xiyuqiao commented 2 years ago

我遇到的类似的报错是: local variable 'stride' referenced before assignment 发现报错的位置上面也有 from thop import profile 安装thop就能成功运行了 pip install thop -i https://pypi.douban.com/simple/

不确定没有成功是不是网上提到的,版本问题,https://blog.csdn.net/qq_37844044/article/details/119644093

ahhhxxx commented 2 years ago

这是来自QQ邮箱的假期自动回复邮件。您好,我最近正在休假中,无法亲自回复您的邮件。我将在假期结束后,尽快给您回复。

yulongnan commented 2 years ago

Thop 版本问题,安装thop-0.0.31.post2005241907-py3-none-any.whl模块包: https://www.cnpython.com/pypi/thop/dl-thop-0.0.31.post2005241907-py3-none-any.whl 直接pip install thop 安装是0.1版本,会报错

ahhhxxx commented 2 years ago

这是来自QQ邮箱的假期自动回复邮件。您好,我最近正在休假中,无法亲自回复您的邮件。我将在假期结束后,尽快给您回复。