![](https://img-blog.csdnimg.cn/20201014180756780.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
_icrazy_
这个作者很懒,什么都没留下…
展开
-
pytorch实现yolo-v3 (源码阅读和复现) -- 001
参考: https://blog.paperspace.com/how-to-implement-a-yolo-object-detector-in-pytorch/ https://blog.paperspace.com/how-to-implement-a-yolo-v3-object-detector-from-scratch-in-pytorch-part-2/ https://ju...翻译 2018-08-10 18:03:45 · 6600 阅读 · 2 评论 -
pytorch实现yolo-v3 (源码阅读和复现) -- 002
上一篇已经介绍了yolov3使用到的网络darknet53每一层的结构,现在这里来完成代码解析和模型创建本章所有代码: ???????????????1. 加载并解析配置文件cfg/yolov3.cfg配置文件包含6种不同type, 分别为'convolutional', 'net', 'route', 'shortcut', 'upsample', 'yolo', 其中'net'...翻译 2018-08-10 18:04:44 · 3485 阅读 · 1 评论 -
pytorch实现yolo-v3 (源码阅读和复现) -- 003算法分析
通过给定锚点在特征图上进行目标位置预测和分类 在上一篇中我们谈到了用于yolo v3 网络模型检测的DetectionLayer层, 它的核心是通过锚点在特征图中进行运算,并通过回归的方式,最终输出目标区域位置坐标和分类信息(yolo v3 目标的分类也是用的回归而不是常用的softmax)功能实现放在了util.py的predict_transform(prediction, inp_di...翻译 2018-08-10 18:06:02 · 2154 阅读 · 0 评论 -
pytorch实现yolo-v3 (源码阅读和复现) -- 004算法分析
对上一步模型直接检测层的预测结果进行进一步过滤, 核心还是nms1.核心代码def write_results(predictions, confidence, num_class, nms=True, nms_thresh=0.4): # 保留预测结果中置信度大于给定阈值的部分 # predictions: shape=(1,10647, 85) # mask...翻译 2018-08-10 18:06:48 · 1546 阅读 · 0 评论 -
pytorch实现yolo-v3 (源码阅读和复现) -- 005
前面已经搭建了YOLO V3的网络, 只需要在此基础上提供输入,并对预测结果过滤,就可以输出了,完整工程代码已分享.所有代码: https://github.com/wanghao00/pytorch-yolo-v3/blob/master/0021. 图像预处理部分preprocess.py给定图像路径以及期望的尺寸, 进行整体缩放后, 居中放置在画布上,并返回, 返回值值包括 :...翻译 2018-08-10 18:07:57 · 2249 阅读 · 3 评论 -
PyTorch的SoftMax交叉熵损失和梯度
在PyTorch中可以方便的验证SoftMax交叉熵损失和对输入梯度的计算示例:注意:官方提供的softmax交叉熵求解结果示例:# -*- coding: utf-8 -*-import torchimport torch.autograd as autogradfrom torch.autograd import Variableimport torch.n...原创 2018-09-14 19:48:48 · 7891 阅读 · 0 评论 -
Pytorch层--AdaptiveAvgPool2d
翻译自: https://discuss.pytorch.org/t/what-is-adaptiveavgpool2d/26897知乎链接: https://www.zhihu.com/question/282046628函数原型: nn.AdaptiveAvgPool2d(output_size)原理解释:更多细节池化模板大小计算(Pool Kernel Size): stenci...翻译 2019-04-26 15:59:59 · 8316 阅读 · 1 评论