yolov5训练结果有问题

Question

yolov5测试图片正常但是

训练图片没有标签🏷
请问大家这是什么原因

Answer 1

让阿豪来帮你解答，本回答参考chatgpt3.5编写提供，如果还有疑问可以评论或留言

根据提供的信息，可能出现以下几种情况：

1. 训练数据集中可能没有正确的标注。因此，在使用训练数据集时，您应该确保图像的标注正确。可以使用标注工具检查训练数据的标注。例如使用LabelImg进行标注。
2. 训练数据集中可能存在文件损坏或丢失。在使用训练数据集之前，请确保数据集中的所有文件都存在且无损坏。
3. 训练数据集中可能存在错误的文件或文件夹命名。确保文件和文件夹命名符合Yolov5的文件命名规则。例如，标注文件和相应的图像文件应该具有相同的文件名。 以上是可能导致无标签或者训练错乱的原因。如果以上问题都不是，可以检查模型是否正确加载，模型参数是否正确。 以下是一个示例代码，可以辅助调试。

# 检查训练数据集中文件是否存在且无损伤
import os
import cv2
data_dir = 'path/to/training/dataset'
for folder in os.listdir(data_dir):
    if not os.path.isdir(os.path.join(data_dir, folder)):
        continue    
    for file in os.listdir(os.path.join(data_dir, folder)):
        if not os.path.isfile(os.path.join(data_dir, folder, file)):
            continue
        filepath = os.path.join(data_dir, folder, file)
        try:
            img = cv2.imread(filepath)
        except:
            print(f'Invalid file found: {filepath}')
# 检查文件命名规则
for folder in os.listdir(data_dir):
    if not os.path.isdir(os.path.join(data_dir, folder)):
        continue    
    for file in os.listdir(os.path.join(data_dir, folder)):
        if not os.path.isfile(os.path.join(data_dir, folder, file)):
            continue
        filepath = os.path.join(data_dir, folder, file)
        annotation_file = os.path.join(data_dir, folder, file.split('.')[0] + '.txt')
        if not os.path.isfile(annotation_file):
            print(f'Missing annotation file for {filepath}')
# 检查模型情况
import torch
from torchvision import models
model = models.resnet18(pretrained=True)
input_tensor = torch.rand(1,3,224,224)
output = model(input_tensor)
print(output.shape)