https://towardsdatascience.com/learning-note-single-shot-multibox-detector-with-pytorch-part-1-38185e84bd79 https://towardsdatascience.com/learning-note-single-shot-multibox-detector-with-pytorch-part-2-dd96bdf4f434 https://towardsdatascience.com/lea…

首先先放下github地址:https://github.com/acm5656/ssd_pytorch 然后放上参考的代码的github地址:https://github.com/amdegroot/ssd.pytorch 为什么要使用pytorch复现呢,因为好多大佬的代码对于萌新真的不友好,看半天看不懂,所以笔者本着学习和练手的目的,尝试复现下,并分享出来帮助其他萌新学习,大佬有兴趣看了后可以提些建议~ 然后对ssd原理感兴趣的同学可以参考我的这篇博客https://www.cnblogs…

论文地址:https://arxiv.org/abs/1512.02325 源码地址:http://github.com/amdegroot/ssd.pytorch 环境1:torch1.9.0+CPU 环境2:torch1.8.1+cu102.torchvision0.9.1+cu102 1. StopIteration.Batch_size设置32,训练至60次报错,训练中断:Batch_size改成8训练至240次报错. 报错原因及解决方法:train.py第165行: # 修改之前 im…

转载自:https://zhuanlan.zhihu.com/p/33544892 前言 目标检测近年来已经取得了很重要的进展,主流的算法主要分为两个类型(参考RefineDet):(1)two-stage方法,如R-CNN系算法,其主要思路是先通过启发式方法(selective search)或者CNN网络(RPN)产生一系列稀疏的候选框,然后对这些候选框进行分类与回归,two-stage方法的优势是准确度高:(2)one-stage方法,如Yolo和SSD,其主要思路是均匀地在图片的不同位置…

项目地址:https://github.com/bharathgs/Awesome-pytorch-list 列表结构: NLP 与语音处理 计算机视觉 概率/生成库 其他库 教程与示例 论文实现 PyTorch 其他项目 自然语言处理和语音处理 该部分项目涉及语音识别.多说话人语音处理.机器翻译.共指消解.情感分类.词嵌入/表征.语音生成.文本语音转换.视觉问答等任务,其中有一些是具体论文的 PyTorch 复现,此外还包括一些任务更广泛的库.工具集.框架. 这些项目有很多是官方的实现,其中…

通过https://github.com/amdegroot/ssd.pytorch,结合论文https://arxiv.org/abs/1512.02325来理解ssd. ssd由三部分组成: base extra predict base原论文里用的是vgg16去掉全连接层. base + extra完成特征提取的功能.得到不同size的feature map,基于这些feature maps,我们再用不同的卷积核去卷积,分别完成类别预测和坐标预测. 基础特征提取网络 特征提取网络由两部分组…

之前,对SSD的论文进行了解读,可以回顾之前的博客:https://www.cnblogs.com/dengshunge/p/11665929.html. 为了加深对SSD的理解,因此对SSD的源码进行了复现,主要参考的github项目是ssd.pytorch. 搭建SSD的项目,可以分成以下四个部分: 数据读取: 网络搭建: 损失函数的构建: 网络测试 接下来,本篇博客重点分析网络测试. 在eval.py文件中,首先需要搭建测试用的网络.此时,需要将传入的第一个参数换成"test"字…

之前,对SSD的论文进行了解读,可以回顾之前的博客:https://www.cnblogs.com/dengshunge/p/11665929.html. 为了加深对SSD的理解,因此对SSD的源码进行了复现,主要参考的github项目是ssd.pytorch. 搭建SSD的项目,可以分成以下三个部分: 数据读取: 网络搭建: 损失函数的构建: 网络测试. 接下来,本篇博客重点分析损失函数的构建. 检测任务的损失函数,与分类任务的损失函数具有很大不同.在检测的损失函数中,不仅需要计类别置信度的差…

之前,对SSD的论文进行了解读,可以回顾之前的博客:https://www.cnblogs.com/dengshunge/p/11665929.html. 为了加深对SSD的理解,因此对SSD的源码进行了复现,主要参考的github项目是ssd.pytorch. 搭建SSD的项目,可以分成以下三个部分: 数据读取: 网络搭建: 损失函数的构建. 网络测试. 接下来,本篇博客重点分析网络搭建. 该部分整体比较简单,思路也很清晰. 首先,在train.py中,网络搭建的函数入口是函数build_ss…

之前,对SSD的论文进行了解读,可以回顾之前的博客:https://www.cnblogs.com/dengshunge/p/11665929.html. 为了加深对SSD的理解,因此对SSD的源码进行了复现,主要参考的github项目是ssd.pytorch. 搭建SSD的项目,可以分成以下三个部分: 数据读取: 网络搭建: 损失函数的构建: 网络测试. 接下来,本篇博客重点分析数据读取. 一.整体框架 SSD的数据读取环节,同样适用于大部分目标检测的环节,具有通用性.为了方便理解,本项目以V…