Emiedon的博客

实际应用时可能比较想获取VGG中间层的输出，那么就可以如下操作：import numpy as npimport torchfrom torchvision import modelsfrom torch.autograd import Variableimport torchvision.transforms as transformsclass CNNShow(): de...

【提特征】RPN网络前面是一个提特征的网络，比如VGG,Res等，传给RPN网络的是一个特征图，其实也就是一个tensor比如用ZF网络（论文里面用的）输出特征图：13*13*256【RPN】拿到模型的特征，RPN网络首先加了一个3*3*256*256的卷积层（其实不是很清楚为什么加，可能是为了扩大感受野）这样就会得到11*11*256的输出，前面11*11是图形矩阵，其中每一个点在原图中都是一个...

最近做一个项目需要修改roi_align和roi_pooling的底层代码，几篇讲的比较好的文章收录一下[PyTorch版的RoIAlign & crop_and_resize中英文]https://javascript.ctolib.com/longcw-RoIAlign-pytorch.htmlhttps://ptorch.com/news/103.html[Pytorch...

代码来源：https://github.com/SirLPS/roi_pooling注释掉的是我工程中需要的代码，没注释的是pytorch版roi_pooling，供各位需要修改roi_pooling的同学参考# -*- coding:UTF-8 -*-import timeimport torchimport torch.nn.functional as Ffrom torch...