如何使用python将文件保存为原始文件名?

ruoxqz4g  于 2021-09-08  发布在  Java
关注(0)|答案(0)|浏览(209)

您好,我想使用与原始图像相同的名称将预测图像保存在文件夹中,但我的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))

暂无答案!

目前还没有任何答案,快来回答吧!

相关问题