Pytorch编程记录
搭建网络的方式:
1.用sequential方式搭建,只能适用于线性网络
2.用forward和init方式搭建
import cv2 from skimage import feature,io from matplotlib import pyplot as plt from service import IMAGE import numpy as np rw = IMAGE() img_path = 'data/zh1.jpg' img_data = rw.read_img(img_path)[-1] img_data = np.swapaxes(img_data,0,1) img_data = np.swapaxes(img_data,1,2) # 转换轴 img_data = np.flip(img_data,axis=2) # 转换波段