COCO Dataset 2018 Stuff Segmentation Challenge
收藏数据集名称
COCO Dataset 2018 Stuff Segmentation Challenge
数据集目的
进行语义分割,特别是针对“stuff”类别的分割任务,旨在推动语义分割技术的最新发展。
数据集格式
- Images Folder: 存放所有训练图像。
- Annotations Folder: 存放对应的地面实况分割图像。
- 注释图像的文件名应与RGB图像的文件名相同。
数据集使用
数据准备
- 安装必要的库:
tensorflow-gpu,tqdm,keras,keras-segmentation。
数据可视化
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"
参考文献




