cv2本地图形读取,由RGB变为灰度,再变为tensor
最近想用pytorch识别本地的图片库,于是打算用一个最基础的网络,没有用到Conv
由cv2读取本地图形方式为:
array1 = cv2.imread("./result/1.jpg", cv2.IMREAD_GRAYSCALE)
x = torch.tensor(array1).float()
第一行选取图形目录,cv2.IMREAD_GRAYSCALE为转为灰度
第二行为转为tensor形式,可以进入计算
基础的网络如下:
myNet = nn.Sequential(
nn.Linear
原创
2020-05-12 16:39:27 ·
1536 阅读 ·
0 评论