pytorch: ModuleNotFoundError: No module named ‘models‘

pytorch: ModuleNotFoundError: No module named ‘models‘_第1张图片
使用yolov5去做检测,正常使用的时候,没有问题。后面因为使用pyqt5编写了一个小demo,使用到yolov5检测的功能,出现的上面问题。
查看原因是文件目录结构的问题
初始启动程序main.py内容
pytorch: ModuleNotFoundError: No module named ‘models‘_第2张图片
更改后的内容
pytorch: ModuleNotFoundError: No module named ‘models‘_第3张图片
main.py函数中添加了:

root_path = os.getcwd()
sys.path.insert(0, root_path + "/application/deepsort/")

/application/deepsort/, 为原始yolov5工程执行检测时,启动的文件路径,也就是models文件夹的位置。文件结构如下:
pytorch: ModuleNotFoundError: No module named ‘models‘_第4张图片

你可能感兴趣的:(pytorch框架内容)