BalancedGroupSoftmax:CVPR 2020口头论文 源码
平衡类Softmax克服分类器不平衡用于长尾物体检测 :warning: 最新:当前的回购是完整版本。 但是我们删除了许多冗余代码,目前仍在测试中。 此回购协议是CVPR 2020口头论文的正式实施:克服带有分类组Softmax的长尾物体检测的分类器不平衡。 注意:当前代码仍然不是很干净。 我们仍在努力中,它将很快得到更新。 要求 1.环境: 要求与。 我们在以下设置上进行了测试: 的Python 3.7 CUDA 9.2 pytorch 1.3.1 + cu92 火炬视觉0.4.2 + cu92 多媒体广告0.2.14 HH=`pwd` conda create -n mmdet python=3.7 -y conda activate mmdet pip install cython pip install numpy pip install torch pip instal
文件列表
BalancedGroupSoftmax:CVPR 2020口头论文
(预估有个269文件)
two_stage.py.back
12KB
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
暂无评论