引言:在训练网络中时常需要将图像格式进行转化,很麻烦,今天记录一下,下面一起来看看吧!!!
1.将numpy----》tensor
废话少说,直接开干!
代码:
import cv2 import torch import numpy as np path=r'./cat.jpg' img=cv2.imread(path) print(img.shape) img1 = torch.from_numpy(img1) print(img1.shape)
#待续。。。
#
2.结束:
GAME OVER
如果理解错误,欢迎大家批评,及时纠正小琼的错误哦,真心感谢你的纠正!!!
如果小伙伴梦有疑问欢迎在评论区留言哦!!!
如果感觉不错的话!点个赞呗,不用客气!(O(∩_∩)O哈哈~)
欢迎和小伙伴梦一起学习,共同努力,加油!!!