yolov5自己训练时图片格式

Question

问题遇到的现象和发生背景

在utils.torch_utils中没有找到load_classifier这个函数模块

问题相关代码，请勿粘贴截图

from numpy import random
import torch
from models.experimental import attempt_load
from utils.dataloaders import LoadStreams, LoadImages
from utils.general import (
    check_img_size, non_max_suppression, apply_classifier, scale_coords,
    xyxy2xywh, strip_optimizer, set_logging)
from utils.torch_utils import select_device, load_classifier

运行结果及报错内容

Traceback (most recent call last):
  File "D:\lj_yolo\yolov5\mydect.py", line 8, in <module>
    from utils.torch_utils import select_device, load_classifier
ImportError: cannot import name 'load_classifier' from 'utils.torch_utils' (D:\lj_yolo\yolov5\utils\torch_utils.py)

进程已结束,退出代码1

我的解答思路和尝试过的方法

去github中查找历史版本找不到相关函数

我想要达到的结果

运行无报错

Answer 1

mydect.py哪里来的，官放版本没有这个东西。
建议直接去下载最新的版本