pawlo2013/one_piece_dataset
收藏数据集卡片 for "one_piece_dataset"
数据集信息
- 特征:
full_colour: 彩色图像,数据类型为图像。sketch: 草图图像,数据类型为图像。
- 分割:
train: 训练集,包含922个样本,总大小为170204480.0字节。
- 下载大小: 170225532字节
- 数据集大小: 170204480.0字节
数据集描述
该数据集包含922张来自《海贼王》动画的图像,每行包含彩色图像和草图图像。
示例设置
以下是数据集的示例设置代码,图像未进行归一化处理: python transform = Compose([ transforms.Resize((128, 128)), transforms.ToTensor(), transforms.Lambda(lambda t: (t * 2) - 1) ])
def transforms(examples): examples["sketch_pixel_values"] = [transform(image.convert("RGB")) for image in examples["sketch"]] examples["full_colour_pixel_values"] = [transform(image.convert("RGB")) for image in examples["full_colour"]] del examples["sketch"] del examples["full_colour"] return examples
dataset = load_dataset("pawlo2013/one_piece_dataset", split="train") transformed_full_colour_dataset = dataset.with_transform(transforms) dataloader = DataLoader(transformed_full_colour_dataset, batch_size=16, shuffle=True, num_workers=0)




