| """ |
| Expected directory format: |
| |
| VideoMatte Train/Valid: |
| ├──fgr/ |
| ├── 0001/ |
| ├── 00000.jpg |
| ├── 00001.jpg |
| ├── pha/ |
| ├── 0001/ |
| ├── 00000.jpg |
| ├── 00001.jpg |
| |
| ImageMatte Train/Valid: |
| ├── fgr/ |
| ├── sample1.jpg |
| ├── sample2.jpg |
| ├── pha/ |
| ├── sample1.jpg |
| ├── sample2.jpg |
| |
| Background Image Train/Valid |
| ├── sample1.png |
| ├── sample2.png |
| |
| Background Video Train/Valid |
| ├── 0000/ |
| ├── 0000.jpg/ |
| ├── 0001.jpg/ |
| |
| """ |
|
|
|
|
| DATA_PATHS = { |
| |
| 'videomatte': { |
| 'train': '../matting-data/VideoMatte240K_JPEG_SD/train', |
| 'valid': '../matting-data/VideoMatte240K_JPEG_SD/valid', |
| }, |
| 'imagematte': { |
| 'train': '../matting-data/ImageMatte/train', |
| 'valid': '../matting-data/ImageMatte/valid', |
| }, |
| 'background_images': { |
| 'train': '../matting-data/Backgrounds/train', |
| 'valid': '../matting-data/Backgrounds/valid', |
| }, |
| 'background_videos': { |
| 'train': '../matting-data/BackgroundVideos/train', |
| 'valid': '../matting-data/BackgroundVideos/valid', |
| }, |
| |
| |
| 'coco_panoptic': { |
| 'imgdir': '../matting-data/coco/train2017/', |
| 'anndir': '../matting-data/coco/panoptic_train2017/', |
| 'annfile': '../matting-data/coco/annotations/panoptic_train2017.json', |
| }, |
| 'spd': { |
| 'imgdir': '../matting-data/SuperviselyPersonDataset/img', |
| 'segdir': '../matting-data/SuperviselyPersonDataset/seg', |
| }, |
| 'youtubevis': { |
| 'videodir': '../matting-data/YouTubeVIS/train/JPEGImages', |
| 'annfile': '../matting-data/YouTubeVIS/train/instances.json', |
| } |
| |
| } |
|
|
