ModuleNotFoundError: No module named ‘models‘解决torch.load问题【天坑】

当使用torch.load时,报错No module named 'models'
在网上查了很多资料说目录结构得和保存时一模一样,话虽如此,但一直没理解要如何一样

因为我是用detect.py调用yolov5的best.pt模型,该模型被自动保存在runs/train/exp/weights/下,但即便改成了这样,还是报同样的错

最后调试发现,No module named 'models'报错说没有models,确实是因为没有才报错的。我用的是ultralytics的yolov5,其结构目录如下
在这里插入图片描述
可以看到,里面真的有两个module:utils和models,即便把detect.py里修改得没有引用这两个module里任何py文件,但在torch.load模型时,还是得把这两个module放在和detect.py同一路径,而.pt模型在哪里都无所谓,最终解决这个bug

花了大概数小时,气死

  • 45
    点赞
  • 75
    收藏
    觉得还不错? 一键收藏
  • 47
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 47
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值