在使用tensorrtx的时候报错:ModuleNotFoundError: No module named ‘utils.torch_utils’
在使用tensorrtx的时候,要进行egine加速,需要将pt模型转化为wts模型,需要在yolo工程下进行
在ultralytics/yolov5.git这个工程下有utils这个文件夹
Traceback (most recent call last):
File “gen_wts.py”, line 6, in
from utils.torch_utils import select_device
ModuleNotFoundError: No module named ‘utils.torch_utils’
https://github.com/ultralytics/yolov5.git