pytorch 使用torch加载经过训练的YoloV5模型

3mpgtkmj  于 2023-06-06  发布在  其他
关注(0)|答案(1)|浏览(299)

我跟随this tutorial使用YoloV5和EasyOCR构建ANPR。在我训练了我的模型之后,我想使用torch加载它,使用以下代码片段:

import torch

yolo = torch.hub.load(
    'ultralytics/yolov5',
    'custom',
    path='/content/yolov5/runs/train/yolov5s_results/weights/best.pt',
    force_reload=True)

但我面对这个错误
ImportError: cannot import name 'is_jupyter' from 'utils.general' (/content/yolov5/utils/general.py)
我真的很感激你能提供的任何帮助。

b09cbbtk

b09cbbtk1#

嗯,奇怪,is_jupyter应该根据this存在。您可以再次尝试git pull,或者删除repo并从头开始完全重新克隆整个内容?我在automatic 1111 stable diffusion webgui中经历过一些git的恶作剧,每当我拉取时,它会更新一些东西,并保持其他东西不变。删除存储库,然后重新克隆解决了我的失踪的东西。也许只是复制repo上www.example.com的内容general.py?如果这还不行,我会按照@ChristophRackwitz的建议和report the issue去做。
如果这是有用的信息,they changed the function from is_notebook to is_jupyter in February,不久前。如果你使用yolov 5的时间超过了这个时间,我真的认为删除你的repo然后从头开始几乎肯定会修复它。看看你的general.py,我想它仍然会有is_notebook功能。

相关问题