Release IDDM v1.1.8-beta.2 · chairc/Integrated-Design-Diffusion-Model

What's Changed

fix: Patch interface post image to solve the MEAN and STD not to work. by @bestl1fe in #104
fix: Fix MEAN and STD bug and update organization logo. by @chairc in #105
chore: MEAN and STD params setting by @chairc in #108
feat: Parameters decomposed into methods; Added PSNR and SSIM calculators; Update requirements.txt. by @chairc in #109
chore: Update model list by @chairc in #110
chore: Add use_gpu params. by @chairc in #111
refactor: Refactor trainer and update README by @chairc in #113
fix: Patch pip and server by @BestChenA in #114
fix: Fix import package safety alerts; Fix the bug that the Flask API could only be called once in the server mode by @chairc in #115
fix: Fix the bug where the pixels of the image exceed 255 or less than 0. by @chairc in #117
feat: Update the short name trigger parameter. by @chairc in #119
docs: Fix neu-120-weight.pt pre-training model download link. by @BestChenA in #121
docs: Update README. by @chairc in #122
docs: Update username. by @BestChenA in #123
docs: Update username. by @chairc in #124

Full Changelog: v1.1.7...v1.1.8-beta.2

Note: The weight include model, ema_model and optimizer.

celebahq-120-weight.pt: Trained on a dataset of 30,000 people face, and image size is 120 (celebahq-120-weight.pt)
animate-ganyu-120-weight.pt: Trained on a dataset of 500 animate ganyu face, and image size is 120 (animate-ganyu-120-weight.pt)
neu-cls-64-weight.pt: Trained on a dataset of 7,226 defect, and image size is 64 (neu-cls-64-weight.pt)
neu-120-weight.pt: Trained on a dataset of 1,800 defect, and image size is 120 (neu-120-weight.pt)
cifar-64-weight.pt: Trained on a dataset of 60,000 images, and image size is 64 (cifar10-64-weight.pt)
animate-face-64-weight.pt: Trained on a dataset of 63,565 animate face, and image size is 64 (animate-face-64-weight.pt)