search

yzqxy / Yolov8_obb_Prune_Track

GNU General Public License v3.0
185 stars 15 forks source link

关于预训练权重和数据集的问题 #11

Open pcycccccc opened 1 year ago

pcycccccc commented 1 year ago

非常感谢大佬提供的代码，我运行了一下是可以跑通的，现有两个问题想要请教一下： 1、我运行的时候载入了yolov8n.pt权重，可能潜意识默认了预训练会更好一些，基本都能识别到目标（单目标），但是框的位置精确性可能没有很好（我标注的数据集可能也存在不够精确的问题，但是还是比检测到的要更准一些）；您提到没有使用预训练权重，是为什么呢？没有载入预训练权重是会检测的更好吗？ 2、我使用的数据集没有包含无目标的图像，想问一下这个代码能否训练包含无目标的图像？

yzqxy commented 1 year ago

非常感谢大佬提供的代码，我运行了一下是可以跑通的，现有两个问题想要请教一下： 1、我运行的时候载入了yolov8n.pt权重，可能潜意识默认了预训练会更好一些，基本都能识别到目标（单目标），但是框的位置精确性可能没有很好（我标注的数据集可能也存在不够精确的问题，但是还是比检测到的要更准一些）；您提到没有使用预训练权重，是为什么呢？没有载入预训练权重是会检测的更好吗？ 2、我使用的数据集没有包含无目标的图像，想问一下这个代码能否训练包含无目标的图像？

1、v8的预训练模型是在coco上训练的，如果你需要做的任务是coco上没有的类别，用预训练模型的作用就不大，载不载入对最后训练的结果没啥影响。 2、可以训练包含无目标的图像，但是如果无目标的图像过多，你的batch-size设置过小，导致某个batch读取的全是无目标的图像时就会报错，这是个小bug，后续需要优化一下

pcycccccc commented 1 year ago

非常感谢大佬提供的代码，我运行了一下是可以跑通的，现有两个问题想要请教一下： 1、我运行的时候载入了yolov8n.pt权重，可能潜意识默认了预训练会更好一些，基本都能识别到目标（单目标），但是框的位置精确性可能没有很好（我标注的数据集可能也存在不够精确的问题，但是还是比检测到的要更准一些）；您提到没有使用预训练权重，是为什么呢？没有载入预训练权重是会检测的更好吗？ 2、我使用的数据集没有包含无目标的图像，想问一下这个代码能否训练包含无目标的图像？

1、v8的预训练模型是在coco上训练的，如果你需要做的任务是coco上没有的类别，用预训练模型的作用就不大，载不载入对最后训练的结果没啥影响。 2、可以训练包含无目标的图像，但是如果无目标的图像过多，你的batch-size设置过小，导致某个batch读取的全是无目标的图像时就会报错，这是个小bug，后续需要优化一下

好的，明白了~多谢您的回答