pre-built detectron2(version = 0.6)
pre-built detectron2(version = 0.6)
网上大部分cityscapes转VOC的代码都不完整,所以写了这个完整转换VOC格式的脚本 注意事项: cityscapes_root需要自行修改,应设为leftImg8bit_trainvaltest和gtFine_trainvaltest所在的目录。 由于cityscapes的中并不包含VOC中的truncated和difficult信息,所以将所有object中truncated和difficult的值都设为0。 cityscapes数据集,通常将val作为test集用于验证性能,因为官方没有放出test集的标注,目的是防止有人用test集训练,在网站上刷分。
使用torchvision中FasterRCNN的模型,dataset使用torchvision中的VOCDetection,数据集使用VOC2012,需要自己修改data_root(即VOCdevkit所在的目录),仅仅进行了一些样本的数据处理。测试过程包含了结果可视化(需安装matplotlib)