Skip to content

Commit

Permalink
Update train_synth_RGB.py
Browse files Browse the repository at this point in the history
  • Loading branch information
VGrondin authored Apr 24, 2023
1 parent c30c52a commit b7bbc4b
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion train_synth_RGB.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",)
Expand Down

0 comments on commit b7bbc4b

Please sign in to comment.