https://github.com/dongdongdong1217/Detectron2-FC
Detectron2-FC是基于Detectron2的神经网络算法快速构建平台。
我们一直在朝着两个方向努力:
(1) 快速搭建复杂深度学习网络算法,并训练与评估;
(2)使(1)中的便捷对几乎所有任务都适用,如分类、预测、目标检测、元学习、半监督、无监督等。
幸运的是,detectron2为我们指明了方向。它是一个非常方便的基于pytorch的算法搭建平台,可以快速复现最新的算法。然而,detectron2只能构建少部分的任务(如目标检测和分割)。例如,简单的图像分类就很难在Detectron2上快速方便地搭建。因此,我们在detectron2的基础上,不断丰富detectron2的功能,使其适用于图像分类、预测、元学习等任务。注意,我们认为我们的工作并不具有创新性,因为只是在Detectron2的架构上,自定义适用于更多任务的函数而已,方便再次使用。我们更希望将这个平台看作一个算法库。
见https://detectron2.readthedocs.io/en/latest/tutorials/install.html. 搭建环境如下:
Ubuntu20.04
CUDA10.1
Pytorch1.8.1
安装好detectron2,去https://github.com/dongdongdong1217/Detectron2-FC下载项目就能使用。