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)             # 转换波段