diff --git a/train_synth_RGB.py b/train_synth_RGB.py index d350b80..aebf62d 100644 --- a/train_synth_RGB.py +++ b/train_synth_RGB.py @@ -331,7 +331,6 @@ def after_step(self): # cfg.merge_from_file(model_zoo.get_config_file("COCO-InstanceSegmentation/mask_rcnn_X_101_32x8d_FPN_3x.yaml")) # cfg.merge_from_file(model_zoo.get_config_file("COCO-Keypoints/keypoint_rcnn_X_101_32x8d_FPN_3x.yaml")) cfg.merge_from_file(model_zoo.get_config_file("COCO-Keypoints/keypoint_rcnn_R_50_FPN_3x.yaml")) - cfg.merge_from_list(opts) cfg.DATASETS.TRAIN = ("tree_train_set",) cfg.DATASETS.VAL = ("tree_val_set",) cfg.DATASETS.TEST = ("tree_test_set",)