IDDM v1.1.0-stable

Note

In this version, The storage format of our model has completely changed, and the model needs to be retrained if the latest code is pulled.
If you don‘t like this version, you can download this tag (v1.0.2).

What's Changed

• Major update: model layer refactored; New network selection (unet and cspdarkunet); Updated the README by @chairc in #11
• Update: Encapsulate code to improve reusability. by @chairc in #12
• Update: Run GPU with custom settings. by @chairc in #13
• Add: Add show and save image in images. by @chairc in #14
• Update: Reconstruct model/networks and other included files. by @chairc in #15
• Update: Add function comments; Update model.parameters() and add SGD optimizer. by @chairc in #16
• Major update: Reconstruct modules.py, and update package path; Rewrite checkpoint storage and loading functions, add checkpoint files by @chairc in #18

New Contributors

• @chairc made their first contribution in #11

Full Changelog: v1.0.2...v1.1.0

About weight model

NEU weight parameters is {"conditional"=True, "image_size"=64, "sample"=ddim, "network"=unet, "act"=gelu, "num_classes"=6}
Cifar10 weight parameters is {"conditional"=True, "image_size"=64, "sample"=ddim, "network"=unet, "act"=gelu, "num_classes"=10}
This weight include model , ema model and optimizer checkpoints.

IDDM v1.0.2

We are delighted to announce that version 1.0.2 of IDDM (Industrial Defect Diffusion Model) has been officially released.
In this version, we have made the following changes.

What's Changed

• Update: Modify tqdm progress bar problem. #b1ea7c3
• Add: Add a parameter for the number of custom visualization images generated. #b48b61b
• Update: Modify the training setting in 4 datasets. #5f20649
• Add: Add fast deployment and API on cloud servers, and send message test function. #f48c468
• Add: Add activation function setting. #8b34de2
• Update: Optimize some code. #cd78a21
• Add: Add saving train log method. #5c4b49a
• Fix: Fix the problem of different training times and modified the distributed training synchronization lock. #c0157d9
• Add: Add selection of activation functions in the generated model. #f8767d2
• Add: Add dataset example. #d3ec97a
• Add: Add results README. #0cd6f57
• Update: Modify README. #526134b
• Add: Add animate face images. #256393c
• Update: Modify the subtitle style. #49dfcf2
• Add: Add the image of IDDM training. #662b4c3
• Update: Add training demo. #8f1ade9
• Update: Modify --distributed initial parameter default value. #6297617
• Fix: Fix the expression type error and changge in to float. #2ca7449
• Update: Modify test error. #4b43d32
• Update: This is a method to display the results of each model during training and can be commented out. #3443662
• Add: Add the 'check_and_create_dir' function. #7f88e91
• Add: Add the activation function in deploy.py. #6529f80
• Fix: Fix no folder error after generate. #5b15de6
• Add: When adding conditional generation, all class images can be generated at once. Multiple images of a single class can also be generated.#a4c77b9
• Update: Add parameter explanation. #dfbe132
• Update: Add model construction explanation. #8a9e121
• Update: Modify model construction explanation error. #b38864c

New Contributors

• @xiaoningpi made their first contribution in #2

Full Changelog: v1.0.1...v1.0.2

IDDM v1.0.1

We are delighted to announce that version 1.0.1 of IDDM (Industrial Defect Diffusion Model) has been officially released.
In this version, we have made the following changes.

What's Changed

• Fix the bug that the model does not save when a single card or CPU is used. #5ee6010
• Fix resuming training model and optimizer parameters loading. #e58cd4e
• Modify the resume training model loading and cancel the broadcast method. #19bc4d9
• Add seed initializer. #9189913
• Add learning rate function such as default, cosine and warmup_cosine. #35d024e
• Add network summary test function. #58f0694
• Modify some formatting issues. #f35040b
• Add requirements.txt #0fb164f
• Refactoring the sampling model structure by creating a base class for common parts and having other sampling methods inherit from it.#278debd
• Add DDIM sample method. #c469e4f
• Add sample selector method during training. #6acf6cc
• Update README #329ded9
• Add image results generated in README. #5df2947
• Modify Repository Structure. #6c21b9b
• Optimizing loading weight method, optimizing generate.py and Check for issues where distributed weight names are not the same as network weights. #24a03c3
• Edited to english in 'train.py' and 'test_module.py'. #788138d
• Edited to english in 'train.py' and 'generate.py'. #f2e9d7c
• Edited to english in 'network.py' and 'modules.py'. #8c0bb55
• Edited to english in 'base.py', 'ddpm.py' and 'ddim.py'. #c92fd82
• Edit english in 'generate.py'. #970ec3c
• Edit untranslated parts in 'utils.py'. #3089bb8

Full Changelog: v1.0.0...v1.0.1

IDDM v1.0.0

We are delighted to announce that version 0.0.1 of IDDM (Industrial Defect Diffusion Model) has been officially released. In this version, we have added a distributed training method, allowing you to make full use of multiple GPUs for larger batch sizes and faster training. Additionally, we have provided pre-trained models on multiple industrial defect datasets, namely NRSD-pretrain and NEUDET-pretrain.

Thank you for your attention, and enjoy using our IDDM!

Full Changelog: v0.0.1...v1.0.0

IDDM v0.0.1alpha

Full Changelog: https://github.com/chairc/Integrated-Design-Diffusion-Model/commits/v0.0.1