improve create_dataset_list.py (#144)

* update all docs involving color label

* modify usage.md

* usage.md

* Update data_prepare.md

* Update jingling2seg.md

* Update jingling2seg.md

* Update labelme2seg.md

* Update jingling2seg.md

* a

* Update config.md

* Update config.md

* Update config.py

* Update config.py

* Update config.py

* Update config.md

* Update basic_group.md

* Update basic_group.md

* Update data_aug.md

* Update config.md

* Update finetune_pspnet.md

* Update finetune_pspnet.md

* Update finetune_unet.md

* Update finetune_unet.md

* Update finetune_icnet.md

* Update finetune_pspnet.md

* Update finetune_hrnet.md

* Update README.md

* Update finetune_deeplabv3plus.md

* Update finetune_deeplabv3plus.md

* Update README.md

* Update config.md

* Update train_group.md

* Update models.md

* Update model_group.md

* Update model_group.md

* Update train_group.md

* Update train_group.md

* Update model_group.md

* Update model_group.md

* Update README.md

* Update README.md

* Update models.md

* Add files via upload

* Add files via upload

* Update models.md

* Update models.md

* Add files via upload

* Update README.md

* Update models.md

* Update models.md

* Update models.md

* Update models.md

* Update models.md

* Update models.md

* Update models.md

* Update models.md

* Update models.md

* Add files via upload

* Update models.md

* Update models.md

* Update loss_select.md

* Update README.md

* Update loss_select.md

* Add files via upload

* Update loss_select.md

* Add files via upload

* Update loss_select.md

* Update README.md

* Add files via upload

* Update models.md

* Add files via upload

* Update models.md

* Update models.md

* Create 1

* Delete 1

* Create 1

* Add files via upload

* Delete 1

* Add files via upload

* Add files via upload

* Update README.md

* Add files via upload

* Update finetune_deeplabv3plus.md

* Update finetune_icnet.md

* Update finetune_pspnet.md

* Update finetune_hrnet.md

* Update finetune_unet.md

* Update config.md

* Update config.md

* Update usage.md

* Update usage.md

* Update usage.md

* Update usage.md

* Update usage.md

* Update usage.md

* Update usage.md

* Update usage.md

* Create download_optic.py

* Add files via upload

* Add files via upload

* Update usage.md

* Update README.md

* Update usage.md

* Update usage.md

* Update usage.md

* add optic.yaml

* Add files via upload

* Delete deeplabv3p_xception65_pet.yaml

* Delete hrnet_w18_pet.yaml

* Delete icnet_pet.yaml

* Delete pspnet.yaml

* Delete unet_pet.yaml

* modify create_dataset_list.py

* modify dataset_list

docs/data_prepare.md

@@ -123,7 +123,7 @@ python pdseg/tools/create_dataset_list.py <your/dataset/dir> ${FLAGS}
 ```
 运行后将在数据集根目录下生成训练/验证/测试集的文件列表（文件主名与`--second_folder`一致，扩展名为`.txt`）。
 
-**Note:** 若训练/验证/测试集缺少标注图片，仍可自动生成不含分隔符和标注图片路径的文件列表。
+**Note:** 生成文件列表要求：要么原图和标注图片数量一致，要么只有原图，没有标注图片。若数据集缺少标注图片，仍可自动生成不含分隔符和标注图片路径的文件列表。
 
 #### 命令行FLAGS列表
 

pdseg/tools/create_dataset_list.py

@@ -116,18 +116,19 @@ def generate_list(args):
         label_files = get_files(1, dataset_split, args)
         if not image_files:
             img_dir = os.path.join(dataset_root, args.folder[0], dataset_split)
-            print("No files in {}".format(img_dir))
+            warnings.warn("No images in {} !!!".format(img_dir))
         num_images = len(image_files)
 
         if not label_files:
             label_dir = os.path.join(dataset_root, args.folder[1], dataset_split)
-            print("No files in {}".format(label_dir))
+            warnings.warn("No labels in {} !!!".format(label_dir))
         num_label = len(label_files)
 
-        if num_images < num_label:
-            warnings.warn("number of images = {}  <  number of labels = {}."
-                          .format(num_images, num_label))
-            continue
+        if num_images != num_label and num_label > 0:
+            raise Exception("Number of images = {}    number of labels = {} \n"
+                            "Either number of images is equal to number of labels, "
+                            "or number of labels is equal to 0.\n"
+                            "Please check your dataset!".format(num_images, num_label))
 
         file_list = os.path.join(dataset_root, dataset_split + '.txt')
         with open(file_list, "w") as f: