圖片保存工具
# 圖片張數
N = 16
# 通道數
C = 3
# 高度
H = 64
# 寬度
W = 32
image = torch.rand(N,C,H,W)
# 路徑
path = './'
save_image(image, path)
圖片保存工具
# 圖片張數
N = 16
# 通道數
C = 3
# 高度
H = 64
# 寬度
W = 32
image = torch.rand(N,C,H,W)
# 路徑
path = './'
save_image(image, path)
本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。