ValueError:找不到在单图像模式下写入指定文件的格式



我正在尝试使用skimage包读取图像,然后裁剪它,然后保存它。修剪前效果良好。保存时,它抛出以下错误

ValueError:找不到写入指定文件的格式单图像模式

下面是我的代码。非常感谢您的帮助。感谢

import os
import numpy as np
import matplotlib.pyplot as plt
import skimage
import dataloader
from utility import To_csv
path='D:\beantech_Data\objtect_detection'
def crop(img):
return skimage.util.crop(img, ((0,500),(0,0),(0,0)))
images, boxes,  labels = dataloader.train_loader(path)
os.makedirs(os.path.join(path, 'train','cropped'), exist_ok=True)
for i in range(len(images)):
croped_image = crop(images[i])
skimage.io.imsave(os.path.join(path, 'train','cropped',f'img{str(i)}'), croped_image)
box = boxes[i]
To_csv(box, i,os.path.join(path, 'train','cropped'), Aug= True )

问题是,代码中没有给出文件格式,即(.png、.jpeg等(。

通过更正这一行,代码运行良好-

skimage.io.imsave(os.path.join(path, 'train','cropped',f'img{str(i)}.png'), croped_image)

感谢

最新更新