数据集:
FredZhang7/anime-prompts-180K
For more info on data collection and the preprocessing algorithm, please see Fast Anime PromptGen .
For disclaimers about the Danbooru data, please see Danbooru Tag Generator .
Download and save the dataset to anime_prompts.csv locally.
pip install datasets
import csv
import datasets
dataset = datasets.load_dataset("FredZhang7/anime-prompts-180K")
train = dataset["train"]
safebooru_clean = train["safebooru_clean"]
danbooru_clean = train["danbooru_clean"]
danbooru_raw = train["danbooru_raw"]
with open("anime_prompts.csv", "w") as f:
writer = csv.writer(f)
writer.writerow(["safebooru_clean", "danbooru_clean", "danbooru_raw"])
for i in range(len(safebooru_clean)):
writer.writerow([safebooru_clean[i], danbooru_clean[i], danbooru_raw[i]])