-
Notifications
You must be signed in to change notification settings - Fork 0
kie4280/DLP-diffusion-model
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
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.
About
DLP lab - Diffusion model
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published