深度学习 学习与训练中遇到的问题 记录

• RuntimeError: Sizes of tensors must match except in dimension 1. Got 61 and 60 in dimension 2

错误产生原因：在模型中有以下操作：torch.cat(x, y)

      当x, y的尺寸不一致时，就会出现以上错误。

      进行单步调试，发现我的错误是因为图像山采样和下采样后，由于取整的方式不一样，导致的x、y的尺寸不一致。

解决方法：

方法1：在图像输入时，宽和高最好是32的整数倍，就不会出现取整的问题。

方法2：把x、y resize到相同的尺寸。

 

• np.newaxis 插入一个新维度 为1

import numpy as np
x=np.arange(4)
print(x)#初始的一维数组
print(x[np.newaxis,:])#得到的二维数组
print(x[:,np.newaxis])#得到的另一个二维数组

[0 1 2 3]
[[0 1 2 3]]
[[0]
 [1]
 [2]
 [3]]

•  pytorch中彩色图像（三通道）转灰度图像（单通道）

from torchvision import transforms

transform = transforms.Compose([
     transforms.Grayscale(num_output_channels=1),#彩色图像转灰度图像，num_output_channels默认1
     transforms.ToTensor()        
])   

2 彩色图转指定RGB通道

def change_image_channels(image):
    #3通道转单通道
    if image.mode == 'RGB':
        r,g,b = image.split()
    return r,g,b

 

• pip install -i 国内镜像

清华源  tsinghua

python3 -m pip install xxx -i https://pypi.tuna.tsinghua.edu.cn/simple

豆瓣源 douban

python3 -m pip install xxx -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

CoCo数据集2014 内容解析 ： https://blog.csdn.net/ai_faker/article/details/106963238

coco数据集地址： https://pjreddie.com/projects/coco-mirror/