这是一个目标检测中选择正负样本的策略。

1.环境准备

ubuntu16.04
cuda10.1
cudnn7
python3.6

git clone https://github.com/sfzhang15/ATSS.git
cd ATSS# because of cuda version error: no AT_CHECK defination
sed -i '16 a #ifndef AT_CHECK \n#define AT_CHECK TORCH_CHECK \n#endif' atss_core/csrc/cuda/deform_pool_cuda.cu
sed -i '13 a #ifndef AT_CHECK \n#define AT_CHECK TORCH_CHECK \n#endif' atss_core/csrc/cuda/deform_conv_cuda.cu

2.模型准备

wget https://drive.google.com/u/0/uc?export=download&confirm=xa89&id=1t8RLdQ6fsFXa0kzPIQ7541uZeQeMXP73

3.demo运行

修改：

vim ./ATSS/atss_core/modeling/rpn/atss/inference.py +46# 修改：
pre_nms_top_n = candidate_inds.view(N, -1).sum(1)#为：
pre_nms_top_n = candidate_inds.reshape(N, -1).sum(1)

本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！