Skip to content

Latest commit

 

History

History
15 lines (10 loc) · 454 Bytes

README.md

File metadata and controls

15 lines (10 loc) · 454 Bytes

add-segmentation-to-coco

Add segmentation masks to COCO detection dataset

Prerequests

  1. SAM weights put to weights dir here - vit_h
  • weights/sam_vit_h_4b8939.pth
  1. COCO json files put to dataset dir
  • dataset/annotation.json
  • dataset/images dir for image files

  1. Install modules pip install -r requirements.txt

  2. Use help with python segmentation_coco.py -h