您好,我想使用与原始图像相同的名称将预测图像保存在文件夹中,但我的cv2.imwrite没有将图像保存在文件夹中。
path2=r'/data/folder/'
def get_segmented(image):
for i, file in enumerate(glob.glob(image)):
print(file,i)
start_time = time()
image = image_loader(file)
output = model(image)
output = output.detach().cpu()
output = np.resize(output, (160, 160))
img_filename = os.path.splitext(file)[0]
print(cv2.imwrite(path2 + img_filename + '.png',output*255))
暂无答案!
目前还没有任何答案,快来回答吧!