修改train.py
这个脚本内涉及一些路径,会导致训练失败,其中以下比较关键
==52行==
config.data_root = os.path.join(config.data_dir, '')
config.annFile = os.path.join(config.data_dir, 'annotations/train.json')
outputs_dir = config.ckpt_path
修改yolo_dataset.py
这个脚本内涉及一些路径,会导致训练失败,其中以下比较关键
==97行== 原始脚本少了如下展示的第三行,会导致找不到图片文件,原脚本img_path
仅获取了图片的名字,97行这部分指向了训练过程的图片文件集
img_id = self.img_ids[img_ids_index]
img_path = coco.loadImgs(img_id)[0]["file_name"]
img_path = "images/train/" + img_path
==178行== 原始脚本少了如下展示的第三行,会导致找不到图片文件,原脚本img_path
仅获取了图片的名字,178行这部分指向了验证过程的图片文件集
img_id = self.img_ids[index]
img_path = coco.loadImgs(img_id)[0]["file_name"]
img_path = "images/val/" + img_path