Open udayallu opened 3 years ago
cfg = get_cfg()
cfg.MODEL.WEIGHTS = "publaynet.pkl" cfg.merge_from_file("faster_rcnn.yaml") cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.7
Non-existent config key: MODEL.TYPE
@udayallu This is a detectron model and you are trying to use it in detectron2 which is not possible.
loading the pre-trained model throwing an error
cfg = get_cfg()
Use the final weights generated after successful training for inference
cfg.MODEL.WEIGHTS = "publaynet.pkl" cfg.merge_from_file("faster_rcnn.yaml") cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.7
Non-existent config key: MODEL.TYPE