Skip to content

earzamastsev/add-segmentation-to-coco

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 

Repository files navigation

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

About

Add segmentation masks to COCO detection dataset

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages