GCNet:GCNet:非本地网络满足挤压激励网络的需求 源码
用于目标检测的GCNet ,,,魏芳云,。 此回购协议是基于open-mmlab的COCO对象检测中的的正式实现。 可以在找到核心运算符GC块。 非常感谢mmdetection的简单简洁的框架。 于2019/10/28更新 GCNet在ICCV 2019 Neural Architects Workshop上获得了最佳论文奖! 于2019/07/01更新 代码被重构。 提供了更多结果,所有配置都可以在configs/gcnet找到。 注意:PyTorch官方SyncBN和Apex SyncBN都有一些稳定性问题。 在训练期间,mAP可能会降至零,并在最后几个时期恢复正常。 于20
文件列表
GCNet:GCNet:非本地网络满足挤压激励网络的需求
(预估有个359文件)
sigmoid_focal_loss_cuda.cu
6KB
sigmoid_focal_loss.cpp
2KB
masked_conv2d_kernel.cu
5KB
masked_conv2d_cuda.cpp
3KB
nms_cpu.cpp
2KB
nms_cuda.cpp
575B
nms_kernel.cu
5KB
roi_pool_kernel.cu
6KB
roi_pool_cuda.cpp
3KB
deform_conv_cuda_kernel.cu
41KB
暂无评论