关于flir数据集推理类别，我训练的时候是三个类别，但将生成权重应用于推理就变成了只有person一类，所有的车辆都错检为person

[HUANG7M]

我将数据集做了处理，xml格式转化为了txt，但是txt也是三个类别没错，训练结果也是三个类别，只有推理有问题，请问是推理代码哪里需要改嘛？

parser.add_argument('--weights', nargs='+', type=str, default='./runs/train/iou/weights/best.pt', help='model.pt path(s)')
parser.add_argument('--source1', type=str, default='./FLIR_ADAS-v5/train/rgb/FLIR_09973_RGB.jpg', help='source') # file/folder, 0 for webcam
parser.add_argument('--source2', type=str, default='./FLIR_ADAS-v5/train/thermal_8_bit/FLIR_09973_PreviewData.jpeg', help='source') # file/folder, 0 for webcam
parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)')
parser.add_argument('--conf-thres', type=float, default=0.4, help='object confidence threshold')
parser.add_argument('--iou-thres', type=float, default=0.45, help='IOU threshold for NMS')
parser.add_argument('--device', default='0', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')
parser.add_argument('--view-img', default=False, action='store_true', help='display results')
parser.add_argument('--save-txt', action='store_true', help='save results to *.txt')
parser.add_argument('--save-conf', action='store_true', help='save confidences in --save-txt labels')
parser.add_argument('--save-crop', action='store_true', help='save cropped prediction boxes')
parser.add_argument('--nosave', action='store_true', help='do not save images/videos')
parser.add_argument('--classes', nargs='+', type=int, default=3, help='filter by class: --class 0, or --class 0 2 3')
我将这里的classes设置为default=3也不行

[HUANG7M]

已解决 是我标签顺序的问题

[zx0v0]

您好，我想请教一下您在FLIR数据集上能不能复现作者的精度呀？我通过作者的链接下载后发现数据集本身还包含一个“dog”分类，想请问您对此有进行处理吗？

[shawynx]

您好，我想请教一下您在FLIR数据集上能不能复现作者的精度呀？我通过作者的链接下载后发现数据集本身还包含一个“dog”分类，想请问您对此有进行处理吗？

你好，请问类dog出现这个问题你有解决吗，我遇到了相同的问题

[Dorablank]

我觉得应该是作者把dog这个类别去掉了吧 该类别在数据集里样本数量仅有108个 除此之外样本数量最少的bicycle也有将近3000个