author is leilei
python 3.5+
tensorflow 1.6
本文借鉴一些论文的思路,但是这些论文均是采用pytorch等动态图深度学习框架。
本人采用tensorflow静态图深度学习框架复现!
主函数写的一般,仅供参考,主要是训练代码部分,未实际训练完测试,因此可能有问题,等以后再用tf时,再改吧!
- 建筑物语义分割数据集: building-segmentation-dataset
- 烟-火目标检测数据集: fire-detect-yolov4