xiezhiyu275

关于训练自己的数据集

1

作者您好，这个是不是没法设置验证集呢

从dataloader里获取数据的时候报错

1

感谢作者，就是我下载了d2sa数据集，并且在builtin.py中设置了数据集的路径 ~~~ _PREDEFINED_SPLITS_D2SA = { "d2sa_train": ("D2SA/images", "D2SA/annotations/D2S_amodal_training_rot0.json"), "d2sa_train_visible": ("D2SA/images", "D2SA/annotations/D2S_amodal_training_rot0.json"), "d2sa_train_aug": ("D2SA/images", "D2SA/annotations/D2S_amodal_augmented.json"), "d2sa_train_aug_visible": ("D2SA/images", "D2SA/annotations/D2S_amodal_augmented.json"), } def register_all_d2sa(root="D:\\Desktop\\refproject\\ShapePrior\\Amodal"): for key, (image_root, json_file) in _PREDEFINED_SPLITS_D2SA.items(): # Assume pre-defined datasets...