aikenml/data_mining

1# Explanation of generator_args is in sam/segment_anything/automatic_mask_generator.py: SamAutomaticMaskGenerator2sam_args = {3    'sam_checkpoint': "ckpt/sam_vit_b_01ec64.pth",4    'model_type': "vit_b",5    'generator_args':{6        'points_per_side': 16,7        'pred_iou_thresh': 0.8,8        'stability_score_thresh': 0.9,9        'crop_n_layers': 1,10        'crop_n_points_downscale_factor': 2,11        'min_mask_region_area': 200,12    },13    'gpu_id': 0,14}15aot_args = {16    'phase': 'PRE_YTB_DAV',17    'model': 'r50_deaotl',18    'model_path': 'ckpt/R50_DeAOTL_PRE_YTB_DAV.pth',19    'long_term_mem_gap': 9999,20    'max_len_long_term': 9999,21    'gpu_id': 0,22}23segtracker_args = {24    'sam_gap': 10, # the interval to run sam to segment new objects25    'min_area': 200, # minimal mask area to add a new mask as a new object26    'max_obj_num': 255, # maximal object number to track in a video27    'min_new_obj_iou': 0.8, # the background area ratio of a new object should > 80% 28}