Hello, if I want to use the pretrained model you provided to test my data sets, is it possible to change the following list to my test data set?
ps：
test1, test2 are my test data sets, their directory structure is consistent with your test data set, train2017, val2017
from：
CLASS2LABEL_MAPPING = {
'real': 0, # 正常图像, MSCOCO, ImageNet等
'ldm-text2im-large-256': 1, # 'CompVis/ldm-text2im-large-256': 'Latent Diffusion', # Latent Diffusion 基础版本
'stable-diffusion-v1-4': 2, # 'CompVis/stable-diffusion-v1-4': 'Stable Diffusion', # 现实版本
'stable-diffusion-v1-5': 3, # 'runwayml/stable-diffusion-v1-5': 'Stable Diffusion', # 现实版本
'stable-diffusion-2-1': 4,
'stable-diffusion-xl-base-1.0': 5,
'stable-diffusion-xl-refiner-1.0': 6,
'sd-turbo': 7,
'sdxl-turbo': 8,
'lcm-lora-sdv1-5': 9,
'lcm-lora-sdxl': 10,
'sd-controlnet-canny': 11,
'sd21-controlnet-canny': 12,
'controlnet-canny-sdxl-1.0': 13,
'stable-diffusion-inpainting': 14,
'stable-diffusion-2-inpainting': 15,
'stable-diffusion-xl-1.0-inpainting-0.1': 16,
}
to：
CLASS2LABEL_MAPPING = {
'real': 0, # 正常图像, MSCOCO, ImageNet等
'test1': 1, # 'CompVis/ldm-text2im-large-256': 'Latent Diffusion', # Latent Diffusion 基础版本
'test2': 2, # 'CompVis/stable-diffusion-v1-4': 'Stable Diffusion', # 现实版本
}
React
Typescript
Django
Laravel
Facebook
Microsoft
Google
Alibaba
D3
Tencent