首页 > 解决方案 > 如何从 .t​​frec 获取 jpeg 或 pngs

问题描述

我是 pytorch 用户,我遇到包含.tfrec的数据,我想将它们转换为 jpeg/png 格式,以便我可以在我的 pytorch 代码中读取它。
我搜索了谷歌,但一无所获。
pytorch 用户如何处理 tfre 的任何帮助

如果我直接阅读它们

import torchvision.transforms as T
from torchvision.datasets import ImageFolder
transform_train = T.Compose([
    T.RandomCrop(128, padding_mode="reflect"),
    T.RandomHorizontalFlip(),
    T.ToTensor()
])
train_ds = ImageFolder(
    root=path_to_folder,
    transform=transform_train
)

它会通过错误

RuntimeError:在子文件夹中找到 0 个文件支持的扩展名是:

.jpg,.jpeg,.png,.ppm,.bmp,.pgm,.tif,.tiff,.webp

标签: tensorflowtfrecord

解决方案


推荐阅读