pytorch实现自定义数据集(Dataset)以VOC数据集为例
VOC数据集(2012)VOC数据集由Annotation(标注)、ImageSets(train.txt、val.txt…)、JPEGImages(原始图像)、SegmentationClass(语义分割标签)、SegmentationObject(实例分割标签)五部分组成。Annotation(标注)文件由.xml文件组成。使用pytorch定义自己的数据集首先需要继承torch.utils.data中的Dataset类。解析xml文件# parse_xml_to_dict 解析xml文件
原创
2022-03-22 17:31:52 ·
2939 阅读 ·
0 评论