COCO Dataset 2018 Stuff Segmentation Challenge
收藏数据集概述
数据集名称
COCO Dataset 2018 Stuff Segmentation Challenge
问题陈述
进行Stuff类别的语义分割。该任务旨在推动Stuff类别语义分割的最新技术水平。
数据集格式
- Images Folder:存放所有训练图像。
- Annotations Folder:存放对应的地面真实分割图像。
- 注释图像的文件名应与RGB图像的文件名相同。
使用方法
数据准备验证
bash
python -m keras_segmentation verify_dataset
--images_path="dataset1/images_prepped_train/"
--segs_path="dataset1/annotations_prepped_train/"
--n_classes=50
数据可视化
bash
python -m keras_segmentation visualize_dataset
--images_path="dataset1/images_prepped_train/"
--segs_path="dataset1/annotations_prepped_train/"
--n_classes=50
模型训练
bash python -m keras_segmentation train --checkpoints_path="path_to_checkpoints" --train_images="dataset1/images_prepped_train/" --train_annotations="dataset1/annotations_prepped_train/" --val_images="dataset1/images_prepped_test/" --val_annotations="dataset1/annotations_prepped_test/" --n_classes=300 --input_height=320 --input_width=640 --model_name="pspnet"
预测
bash python -m keras_segmentation predict --checkpoints_path="path_to_checkpoints" --input_path="dataset1/images_prepped_test/" --output_path="path_to_predictions"
参考文献




