Trong-An (Andrew) Bui trongan93
trongan93
/ rareplane_segmentation_1_7
Created
April 9, 2021 05:47
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| cfg.MODEL.WEIGHTS = os.path.join(cfg.OUTPUT_DIR, "model_final.pth") # path to the model we just trained | |
| cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.7 # set a custom testing threshold | |
| predictor = DefaultPredictor(cfg) |
trongan93
/ rareplane_segmentation_1_7
Created
April 9, 2021 05:49
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| from detectron2.utils.visualizer import ColorMode | |
| json_file_test = "/mnt/d/RarePlanes/datasets/synthetic/metadata_annotations/instances_test_aircraft.json" | |
| coco_test=COCO(json_file_test) | |
| catIds = coco_test.getCatIds(catNms=['aircraft']); | |
| imgIds = coco_test.getImgIds(catIds=catIds); | |
| img = coco_test.loadImgs(imgIds[np.random.randint(0,len(imgIds))])[0] | |
| im = cv2.imread("/mnt/d/RarePlanes/datasets/synthetic/test/images/" + img['file_name']) | |
| plt.axis('off') |
OlderNewer