- 将粗糙的高层语义特征和细粒度的底层特征融合。
- download the pretrain model of resnet_v1_101.ckpt
- download the dataset of pascal voc dataset
- download the RefineNet model,
eragonruan
trained on pascal voc
-
直接执行的文件说明(按顺序执行即可):
build_color_map.py
得到类别和颜色对应的字典convert_pascal_voc_to_tfrecords.py
将数据变成tfrecord格式train.py
训练,读取数据并进行训练demo.py
测试,对test_data_path
的所有图片进行测试
-
其它文件说明:
model.py
模型,定义了RefineNetresnet
残差网络utils
工具函数augmentation.py
训练时对数据进行增强pascal_voc.py
获取voc图片pathtfrecords.py
对数据进行转换training.py
训练时处理label的相关函数visualization.py
对测试结果进行可视化