HUANG7M

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

5

我将数据集做了处理，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...