| def generate_sam_args(sam_checkpoint="ckpt", model_type="vit_b", points_per_side=16, | |
| pred_iou_thresh=0.8, stability_score_thresh=0.9, crop_n_layers=1, | |
| crop_n_points_downscale_factor=2, min_mask_region_area=200, gpu_id=0): | |
| sam_args = { | |
| 'sam_checkpoint': f'{sam_checkpoint}/{model_type}.pth', | |
| 'model_type': model_type, | |
| 'generator_args': { | |
| 'points_per_side': points_per_side, | |
| 'pred_iou_thresh': pred_iou_thresh, | |
| 'stability_score_thresh': stability_score_thresh, | |
| 'crop_n_layers': crop_n_layers, | |
| 'crop_n_points_downscale_factor': crop_n_points_downscale_factor, | |
| 'min_mask_region_area': min_mask_region_area, | |
| }, | |
| 'gpu_id': gpu_id} | |
| return sam_args | |