- 博客(6)
- 资源 (2)
- 收藏
- 关注
原创 pytorch之torch.zeros_like
torch.zeros_like:生成和括号内变量维度维度一致的全是零的内容。import torcha = torch.rand(5,1)print(a)n=torch.zeros_like(a)print('n=',n)tensor([[0.9653], [0.5581], [0.1648], [0.3715], [0.2194]])n= tensor([[0.], [0.], [0..
2020-07-27 15:36:52 35590
原创 pytorch 利用netron可视化网络结构
if __name__ == '__main__': import torch.onnx import netron#####hourglass是自己的网络代码,自己定义的网络结构类名 pose = HourglassNet(Bottleneck, 256, 8, 1, 16) # .cuda() dummy_input = torch.randn(1, 3,256, 256)#输出的文件名称,一般是在当前定义网络路径下 onnx_path= "pose.o.
2020-07-27 13:12:51 627
原创 pytorch之resnet中def _make_residual解释layers
def _make_residual(self, block, planes, num_blocks, stride=1): downsample = None if stride != 1 or self.inplanes != planes * block.expansion: downsample = nn.Sequential( nn.Conv2d(self.inplanes, planes * bloc...
2020-07-23 10:58:42 929
原创 pytorch利用tensorboardx可视化网络结构图
1、对于自己网络有外部参数需要输入时:如下所示,外部参数有:nstack, inp_dim, oup_dim ,bnclass PoseNet(nn.Module): def __init__(self, nstack, inp_dim, oup_dim, bn=False, increase=128, init_weights=True, **kwargs):2、定义可视化结构 1)首先,对网络PoseNet传入参数输入:如,nstack=4; inp_dim=256;......
2020-07-22 11:16:52 1135
原创 NMS(非极大值抑制)
近段时间一直在看multi-person pose estimation,在最后有一个剔除候选keypoint的过程,该过程采用的NMS方法。但我一直对NMS不理解,经过查找资料,整理出本人对NMS的理解内容,如有不对请指出。在引进NMS概念之前我们先介绍IOU(交并比),因为在NMS中选出框图起到至关重要的作用1、IOU(交并比) 物体检测需要定位出物体的bounding box,就像上面的图片一样,我们不仅要定位出车辆的bounding...
2020-07-17 10:09:25 42098 13
转载 Ubuntu 16.04通过命令行连接Wi-Fi
https://blog.csdn.net/david394/article/details/106783963
2020-07-13 10:31:40 431
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人