no coco

Signed-off-by: lizz <lizz@sensetime.com>
open-mmlab · Feb 19, 2021 · ddee001 · ddee001
1 parent 8677006
commit ddee001
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/tools/data/matting/comp1k/README.md b/tools/data/matting/comp1k/README.md
@@ -97,7 +97,7 @@ python tools/data/matting/comp1k/preprocess_comp1k_dataset.py data/adobe_composi
 Extend the foreground of training set as follows:
 
 ```shell
-python tools/data/matting/comp1k/extend_fg.py data/adobe_composition-1k data/coco
+python tools/data/matting/comp1k/extend_fg.py data/adobe_composition-1k
 ```
 
 ## Check Directory Structure for DIM

diff --git a/tools/data/matting/comp1k/extend_fg.py b/tools/data/matting/comp1k/extend_fg.py
@@ -80,7 +80,6 @@ def parse_args():
         description='Prepare Adobe composition 1k dataset',
         formatter_class=argparse.ArgumentDefaultsHelpFormatter)
     parser.add_argument('data_root', help='Adobe composition 1k dataset root')
-    parser.add_argument('coco_root', help='COCO2014 or COCO2017 dataset root')
     parser.add_argument(
         '--nproc', type=int, default=4, help='number of processer')
     args = parser.parse_args()
@@ -91,8 +90,6 @@ def main():
     args = parse_args()
     if not osp.exists(args.data_root):
         raise FileNotFoundError(f'{args.data_root} does not exist!')
-    if not osp.exists(args.coco_root):
-        raise FileNotFoundError(f'{args.coco_root} does not exist!')
 
     data_root = args.data_root