pytorch
文章平均质量分 77
init_bin
路上…
展开
-
Pytorch: conv2d、空洞卷积、maxpool2d、 ConvTranspose2d的输出特征图计算方式
1、conv2dnn.Conv2d(in_dim, out_dim, kernel_size=3, stride=1, padding=1)输入为(h1,w1),输出为(h2,w2),h2的计算如下,w2同理:pytorch中如果是3*3的卷积,且步长为1,此时padding设置为1,那么输出特征图大小和输入特征图大小相同。2、空洞卷积——conv2dnn.Conv2d(in_dim, out_dim, kernel_size=3, stride=1, padding=2, di原创 2021-04-14 11:18:22 · 4006 阅读 · 10 评论 -
pytorch调用预训练模型
1.pytorch提供以下模型AlexNet: AlexNet variant from the “One weird trick” paper. VGG: VGG-11, VGG-13, VGG-16, VGG-19 (with and without batch normalization) ResNet: ResNet-18, ResNet-34, ResNet-50, ResNet...原创 2019-08-20 22:41:03 · 5526 阅读 · 3 评论 -
pytorch与tensorflow的numpy与tensor互转
一、tensorflow的numpy与tensor互转1.数组(numpy)转tensor利用tf.convert_to_tensor(numpy),将numpy转成tensor>>> a = np.ones((3,3))>>> aarray([[1., 1., 1.], [1., 1., 1.], [1., 1.,...原创 2019-09-17 10:44:24 · 2672 阅读 · 0 评论