想把网络跑出来的图像放入matlab中处理(好吧因为不知该怎么在python中将bgr->rgb),但是在matlab中批量处理,文件名我不想改变。
所以可以从下面代码来操作:
name_list =[]
for filename in os.listdir(path=r'test_image/ori'):
name_list.append(filename)
for test_image in enumerate(test_dataloader):
idx = test_image[0] # 这里返回的是第idx张图片
...
output = net(input)
output_name = name_list[idx]
save_image(output, output_name)
这样就能保持文件名不变的保存生成的图像了!