【目标检测】ATSS复现

拼搏现实的明天。 2022-09-07 01:19 212阅读 0赞

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

# 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)

想要保存结果的话，可以在demo/atss\_demo.py的104行添加如下代码：

save_path = os.path.join(args.images_dir, "save")
    if not os.path.exists(save_path):
        os.makedirs(save_path)
    cv2.imwrite(os.path.join(save_path, im_name), composite)

运行demo：

python demo/atss_demo.py

结果如下：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM1OTc1NDQ3_size_16_color_FFFFFF_t_70][]

#  4.后续 #

后续有时间的话，看下怎么加到我们自己的模型中去训练。

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM1OTc1NDQ3_size_16_color_FFFFFF_t_70]: /images/20220829/017283b5fb4e46479fc054382764144c.png

发表评论取消回复

表情：

评论列表（有 0 条评论，212人围观）

还没有评论，来说两句吧...

相关阅读

相关睿智的目标检测6——yolo2详解及其预测代码复现

睿智的目标检测6——yolo2详解及其预测代码复现学习前言代码下载实现思路 1、yolo2的预测思路（网络构建思路） 2

痛定思痛。/ 2023年05月31日 03:17/ 0 赞/ 12 阅读

相关【多目标跟踪】FairMOT复现 (CVPR2020)

代码：[https://github.com/ifzhang/FairMOT][https_github.com_ifzhang_FairMOT] 论文：[https:

超、凢脫俗/ 2023年02月18日 08:20/ 0 赞/ 439 阅读

相关【行为检测】G-TAD复现（CVPR 2020）

其实作者给出的环境很详细。这里只是为了记录下环境。便于以后傻瓜复现。代码参考： [https://github.com/Frostinassiky/gtad][https

待我称王封你为后i/ 2023年01月11日 08:48/ 0 赞/ 177 阅读

相关论文阅读|目标检测之指出anchor-free和anchor-based本质区别的ATSS

目录论文相关信息摘要 1 .介绍 2 .相关工作 3 . 分析anchor-based 和anchor-free检测器的不同

落日映苍穹つ/ 2023年01月10日 10:28/ 0 赞/ 132 阅读

相关【人脸检测】SRN测试与评估复现

参考： [https://github.com/ChiCheng123/SRN][https_github.com_ChiCheng123_SRN] 0.环境

Myth丶恋晨/ 2022年12月24日 13:54/ 0 赞/ 151 阅读

相关【人脸检测】复现Pytorch_Retinaface(Pytorch版本)

[https://github.com/biubug6/Pytorch\_Retinaface][https_github.com_biubug6_Pytorch_Retina

淩亂°似流年/ 2022年11月21日 14:58/ 0 赞/ 290 阅读

相关【人脸检测】复现官方RetinaFace

[https://github.com/deepinsight/insightface/tree/master/detection/RetinaFace][https_gith

刺骨的言语ヽ痛彻心扉/ 2022年11月21日 09:51/ 0 赞/ 248 阅读

相关精读目标检测5——yolo2详解及其预测代码复现

学习前言 ……最近在学习yolo1、yolo2和yolo3，写这篇博客主要是为了让自己对yolo2的结构有更加深刻的理解，同时要理解清楚先验框的含义。 ![在这里插入

Bertha 。/ 2022年11月17日 13:51/ 0 赞/ 257 阅读

相关【目标跟踪】Yolov5_DeepSort_Pytorch复现

1.环境 ubuntu16.04 cuda10.1 cudnn7 python3.6 Cython m

蔚落/ 2022年11月02日 11:40/ 0 赞/ 331 阅读

相关【目标检测】ATSS复现

这是一个目标检测中选择正负样本的策略。 1.环境准备 ubuntu16.04 cuda10.1 cudnn7 python3.6

拼搏现实的明天。/ 2022年09月07日 01:19/ 0 赞/ 213 阅读