kie4280 / dlp-diffusion-model Goto Github PK

This is the specification file for conditional DDPM training and testing dataset - i-CLEVR

1. objects.json
This file is a dictionary file that contains the number of objects and the idexes.
There are totally 24 objects in i-CLEVR datasets with 3 shapes and 8 colors.

2. train.json
The file is for training. The number off training data is 18009.
train.json is a dictionary where keys are filenames and values are objects/
For example:
{"CLEVR_train_001032_0.png": ["yellow sphere"], 
"CLEVR_train_001032_1.png": ["yellow sphere", "gray cylinder"], 
"CLEVR_train_001032_2.png": ["yellow sphere", "gray cylinder", "purple cube"], ... }
One image can include objects from 1 to 3

3. test.json, new_test.json
The file are for testing. The number of testing data is 32 for each file.
Both file contain a list where each element includes multiple objects
For example:
[['gray cube'],
['red cube'],
['blue cube'],
['blue cube', 'green cube'], ...]

4. evlauator.py, checkpoint.pth
Here are the pretrained classifier model's script and its weight for evaluating your synthetic images.
More detail can be found in the script file.