2 Star 1 Fork 6

lifw88/RobustVideoMatting

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
train_config.py 1.74 KB
一键复制 编辑 原始数据 按行查看 历史
Peter Lin 提交于 2021-06-12 07:50 . Official code release
"""
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/train',
},
'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',
}
}
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/lifw88/RobustVideoMatting.git
git@gitee.com:lifw88/RobustVideoMatting.git
lifw88
RobustVideoMatting
RobustVideoMatting
master

搜索帮助