息肉定位:使用单发多盒检测器在结肠镜检查视频中进行息肉定位 源码

有关设置环境,请阅读SETUP.md 息肉定位和检测 这项工作是在EEML暑期学校(2019年)期间介绍的。在 查找随附的海报。 对于我们的定位和检测任务,我们使用了ColonCVC数据集和ETIS-Larib数据集。 1.息肉本地化 这里的任务是训练完全卷积网络(FCN-8)为息肉创建分割蒙版,然后在其周围绘制一个边界框。 要训​​练FCN-8,请运行main.py 要评估/测试模型,请运行predict_masks.py 2.息肉检测 这里的任务是训练物体检测网络。 我们使用SSD(单发Multibox检测器评估我们的结果)。 代码适用于Faster R-CNN和SSD 。 要训​​练