Skip to content

Commit 66f033d

Browse files
chaircchairc
committed
2 parents 5c40a6a + 4e836d6 commit 66f033d

49 files changed

Lines changed: 660 additions & 516 deletions

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

‎README.md‎

Lines changed: 51 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -1,30 +1,57 @@
11
<div align="center">
2+
<p>
3+
<a href="https://github.com/chairc/Integrated-Design-Diffusion-Model">
4+
<img width="242" height="121" src="./assets/iddm-logo.png" alt="IDDM">
5+
</a>
6+
</p>
7+
<h1>IDDM: Integrated Design Diffusion Model</h1>
8+
<p>English | <a href="./README_zh.md">简体中文</a></p>
29

3-
<p>
4-
<a href="https://github.com/chairc/Integrated-Design-Diffusion-Model">
5-
<img width="242" height="121" src="./assets/iddm-logo.png" alt="IDDM">
6-
</a>
7-
</p>
8-
9-
# IDDM: Integrated Design Diffusion Model
10-
11-
English | [简体中文](README_zh.md)
12-
13-
[![IDDM](https://img.shields.io/badge/IDDM-Integrated%20Design%20Diffusion%20Model-blue.svg)](https://github.com/chairc/Integrated-Design-Diffusion-Model)
14-
[![DOI](https://zenodo.org/badge/DOI/10.5281/zenodo.10866128.svg)](https://doi.org/10.5281/zenodo.10866128)
15-
[![DeepWiki](https://deepwiki.com/badge.svg)](https://deepwiki.com/chairc/Integrated-Design-Diffusion-Model)
10+
</div>
1611

17-
[![License](https://img.shields.io/badge/license-Apache_2.0-green)](./LICENSE)
18-
[![Issues](https://img.shields.io/github/issues/chairc/Integrated-Design-Diffusion-Model.svg)](https://github.com/chairc/Integrated-Design-Diffusion-Model/issues)
19-
[![Release](https://img.shields.io/github/v/release/chairc/Integrated-Design-Diffusion-Model)](https://github.com/chairc/Integrated-Design-Diffusion-Model/releases)
20-
[![Downloads](https://img.shields.io/github/downloads/chairc/Integrated-Design-Diffusion-Model/total?color=3eb370)](https://img.shields.io/github/downloads/chairc/Integrated-Design-Diffusion-Model/total?color=3eb370)
12+
<div align="center">
13+
<a href="https://github.com/chairc/Integrated-Design-Diffusion-Model" target="_blank">
14+
<img src="https://img.shields.io/badge/IDDM-Integrated Design Diffusion Model-blue.svg" alt="IDDM">
15+
</a>
16+
<a href="https://doi.org/10.5281/zenodo.10866128">
17+
<img src="https://zenodo.org/badge/DOI/10.5281/zenodo.10866128.svg" alt="DOI">
18+
</a>
19+
<a href="https://deepwiki.com/chairc/Integrated-Design-Diffusion-Model" target="_blank">
20+
<img src="https://deepwiki.com/badge.svg" alt="DeepWiki">
21+
</a>
22+
</div>
2123

22-
[![Github star](https://img.shields.io/github/stars/chairc/Integrated-Design-Diffusion-Model.svg)](https://github.com/chairc/Integrated-Design-Diffusion-Model/stargazers)
23-
[![Github fork](https://img.shields.io/github/forks/chairc/Integrated-Design-Diffusion-Model?color=eb6ea5)](https://github.com/chairc/Integrated-Design-Diffusion-Model/forks)
24-
[![Gitee star](https://gitee.com/chairc/Integrated-Design-Diffusion-Model/badge/star.svg?theme=blue)](https://gitee.com/chairc/Integrated-Design-Diffusion-Model)
25-
[![Gitee fork](https://gitee.com/chairc/Integrated-Design-Diffusion-Model/badge/fork.svg?theme=blue)](https://gitee.com/chairc/Integrated-Design-Diffusion-Model)
26-
[![GitCode star](https://gitcode.com/chairc/Integrated-Design-Diffusion-Model/star/badge.svg)](https://gitcode.com/chairc/Integrated-Design-Diffusion-Model)
24+
<div align="center">
25+
<a href="https://github.com/chairc/Integrated-Design-Diffusion-Model/blob/main/LICENSE" target="_blank">
26+
<img src="https://img.shields.io/github/license/chairc/Integrated-Design-Diffusion-Model" alt="License">
27+
</a>
28+
<a href="https://github.com/chairc/Integrated-Design-Diffusion-Model/issues" target="_blank">
29+
<img src="https://img.shields.io/github/issues/chairc/Integrated-Design-Diffusion-Model.svg" alt="GitHub issues">
30+
</a>
31+
<a href="https://github.com/chairc/Integrated-Design-Diffusion-Model/releases" target="_blank">
32+
<img src="https://img.shields.io/github/v/release/chairc/Integrated-Design-Diffusion-Model" alt="GitHub release">
33+
</a>
34+
<a href="#" target="_blank">
35+
<img src="https://img.shields.io/github/downloads/chairc/Integrated-Design-Diffusion-Model/total?color=3eb370" alt="GitHub Downloads">
36+
</a>
37+
</div>
2738

39+
<div align="center">
40+
<a href="https://github.com/chairc/Integrated-Design-Diffusion-Model/stargazers" target="_blank">
41+
<img src="https://img.shields.io/github/stars/chairc/Integrated-Design-Diffusion-Model.svg" alt="GitHub star">
42+
</a>
43+
<a href="https://github.com/chairc/Integrated-Design-Diffusion-Model/forks" target="_blank">
44+
<img src="https://img.shields.io/github/forks/chairc/Integrated-Design-Diffusion-Model?color=eb6ea5" alt="GitHub forks">
45+
</a>
46+
<a href="https://gitee.com/chairc/Integrated-Design-Diffusion-Model" target="_blank">
47+
<img src="https://gitee.com/chairc/Integrated-Design-Diffusion-Model/badge/star.svg?theme=blue" alt="Gitee star">
48+
</a>
49+
<a href="https://gitee.com/chairc/Integrated-Design-Diffusion-Model" target="_blank">
50+
<img src="https://gitee.com/chairc/Integrated-Design-Diffusion-Model/badge/fork.svg?theme=blue" alt="Gitee forks">
51+
</a>
52+
<a href="https://gitcode.com/chairc/Integrated-Design-Diffusion-Model" target="_blank">
53+
<img src="https://gitcode.com/chairc/Integrated-Design-Diffusion-Model/star/badge.svg" alt="GitCode star">
54+
</a>
2855
</div>
2956

3057

@@ -71,7 +98,7 @@ The following packages are required.
7198
coloredlogs==15.0.1
7299
gradio>=5.31.0
73100
matplotlib==3.7.1
74-
numpy==1.25.0
101+
numpy>=1.25.0
75102
Pillow==10.3.0
76103
Requests>=2.32.4
77104
scikit-image==0.22.0
@@ -133,6 +160,7 @@ pip install . # Or python setup.py install
133160
- [x] [2025-08-01] Adding LDM Method. **Support generate 512*512 images** and use Latent Diffusion and reduce GPU memory usage.
134161
- [x] [2025-08-21] Support flash-attn, fast and GPU memory-efficient.
135162
- [x] [2025-10-01] Adding the deployment of Docker file.
163+
- [x] [2025-12-16] Adding custom logger mod.
136164
- [ ] [To be determined] Refactor the project by Baidu PaddlePaddle.
137165

138166

‎README_zh.md‎

Lines changed: 51 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -1,30 +1,57 @@
11
<div align="center">
2+
<p>
3+
<a href="https://github.com/chairc/Integrated-Design-Diffusion-Model">
4+
<img width="242" height="121" src="./assets/iddm-logo.png" alt="IDDM">
5+
</a>
6+
</p>
7+
<h1>IDDM: Integrated Design Diffusion Model</h1>
8+
<p><a href="./README.md">English</a> | 简体中文</p>
29

3-
<p>
4-
<a href="https://github.com/chairc/Integrated-Design-Diffusion-Model">
5-
<img width="242" height="121" src="./assets/iddm-logo.png" alt="IDDM">
6-
</a>
7-
</p>
8-
9-
# IDDM: Integrated Design Diffusion Model
10-
11-
[English](README.md) | 简体中文
12-
13-
[![IDDM](https://img.shields.io/badge/IDDM-Integrated%20Design%20Diffusion%20Model-blue.svg)](https://github.com/chairc/Integrated-Design-Diffusion-Model)
14-
[![DOI](https://zenodo.org/badge/DOI/10.5281/zenodo.10866128.svg)](https://doi.org/10.5281/zenodo.10866128)
15-
[![zread](https://img.shields.io/badge/Ask_Zread-_.svg?style=flat&color=00b0aa&labelColor=000000&logo=data%3Aimage%2Fsvg%2Bxml%3Bbase64%2CPHN2ZyB3aWR0aD0iMTYiIGhlaWdodD0iMTYiIHZpZXdCb3g9IjAgMCAxNiAxNiIgZmlsbD0ibm9uZSIgeG1sbnM9Imh0dHA6Ly93d3cudzMub3JnLzIwMDAvc3ZnIj4KPHBhdGggZD0iTTQuOTYxNTYgMS42MDAxSDIuMjQxNTZDMS44ODgxIDEuNjAwMSAxLjYwMTU2IDEuODg2NjQgMS42MDE1NiAyLjI0MDFWNC45NjAxQzEuNjAxNTYgNS4zMTM1NiAxLjg4ODEgNS42MDAxIDIuMjQxNTYgNS42MDAxSDQuOTYxNTZDNS4zMTUwMiA1LjYwMDEgNS42MDE1NiA1LjMxMzU2IDUuNjAxNTYgNC45NjAxVjIuMjQwMUM1LjYwMTU2IDEuODg2NjQgNS4zMTUwMiAxLjYwMDEgNC45NjE1NiAxLjYwMDFaIiBmaWxsPSIjZmZmIi8%2BCjxwYXRoIGQ9Ik00Ljk2MTU2IDEwLjM5OTlIMi4yNDE1NkMxLjg4ODEgMTAuMzk5OSAxLjYwMTU2IDEwLjY4NjQgMS42MDE1NiAxMS4wMzk5VjEzLjc1OTlDMS42MDE1NiAxNC4xMTM0IDEuODg4MSAxNC4zOTk5IDIuMjQxNTYgMTQuMzk5OUg0Ljk2MTU2QzUuMzE1MDIgMTQuMzk5OSA1LjYwMTU2IDE0LjExMzQgNS42MDE1NiAxMy43NTk5VjExLjAzOTlDNS42MDE1NiAxMC42ODY0IDUuMzE1MDIgMTAuMzk5OSA0Ljk2MTU2IDEwLjM5OTlaIiBmaWxsPSIjZmZmIi8%2BCjxwYXRoIGQ9Ik0xMy43NTg0IDEuNjAwMUgxMS4wMzg0QzEwLjY4NSAxLjYwMDEgMTAuMzk4NCAxLjg4NjY0IDEwLjM5ODQgMi4yNDAxVjQuOTYwMUMxMC4zOTg0IDUuMzEzNTYgMTAuNjg1IDUuNjAwMSAxMS4wMzg0IDUuNjAwMUgxMy43NTg0QzE0LjExMTkgNS42MDAxIDE0LjM5ODQgNS4zMTM1NiAxNC4zOTg0IDQuOTYwMVYyLjI0MDFDMTQuMzk4NCAxLjg4NjY0IDE0LjExMTkgMS42MDAxIDEzLjc1ODQgMS42MDAxWiIgZmlsbD0iI2ZmZiIvPgo8cGF0aCBkPSJNNCAxMkwxMiA0TDQgMTJaIiBmaWxsPSIjZmZmIi8%2BCjxwYXRoIGQ9Ik00IDEyTDEyIDQiIHN0cm9rZT0iI2ZmZiIgc3Ryb2tlLXdpZHRoPSIxLjUiIHN0cm9rZS1saW5lY2FwPSJyb3VuZCIvPgo8L3N2Zz4K&logoColor=ffffff)](https://zread.ai/chairc/Integrated-Design-Diffusion-Model)
10+
</div>
1611

17-
[![License](https://img.shields.io/badge/license-Apache_2.0-green)](./LICENSE)
18-
[![Issues](https://img.shields.io/github/issues/chairc/Integrated-Design-Diffusion-Model.svg)](https://github.com/chairc/Integrated-Design-Diffusion-Model/issues)
19-
[![Release](https://img.shields.io/github/v/release/chairc/Integrated-Design-Diffusion-Model)](https://github.com/chairc/Integrated-Design-Diffusion-Model/releases)
20-
[![Downloads](https://img.shields.io/github/downloads/chairc/Integrated-Design-Diffusion-Model/total?color=3eb370)](https://img.shields.io/github/downloads/chairc/Integrated-Design-Diffusion-Model/total?color=3eb370)
12+
<div align="center">
13+
<a href="https://github.com/chairc/Integrated-Design-Diffusion-Model" target="_blank">
14+
<img src="https://img.shields.io/badge/IDDM-Integrated Design Diffusion Model-blue.svg" alt="IDDM">
15+
</a>
16+
<a href="https://doi.org/10.5281/zenodo.10866128">
17+
<img src="https://zenodo.org/badge/DOI/10.5281/zenodo.10866128.svg" alt="DOI">
18+
</a>
19+
<a href="https://zread.ai/chairc/Integrated-Design-Diffusion-Model" target="_blank">
20+
<img src="https://img.shields.io/badge/Ask_Zread-_.svg?style=flat&color=00b0aa&labelColor=000000&logo=data%3Aimage%2Fsvg%2Bxml%3Bbase64%2CPHN2ZyB3aWR0aD0iMTYiIGhlaWdodD0iMTYiIHZpZXdCb3g9IjAgMCAxNiAxNiIgZmlsbD0ibm9uZSIgeG1sbnM9Imh0dHA6Ly93d3cudzMub3JnLzIwMDAvc3ZnIj4KPHBhdGggZD0iTTQuOTYxNTYgMS42MDAxSDIuMjQxNTZDMS44ODgxIDEuNjAwMSAxLjYwMTU2IDEuODg2NjQgMS42MDE1NiAyLjI0MDFWNC45NjAxQzEuNjAxNTYgNS4zMTM1NiAxLjg4ODEgNS42MDAxIDIuMjQxNTYgNS42MDAxSDQuOTYxNTZDNS4zMTUwMiA1LjYwMDEgNS42MDE1NiA1LjMxMzU2IDUuNjAxNTYgNC45NjAxVjIuMjQwMUM1LjYwMTU2IDEuODg2NjQgNS4zMTUwMiAxLjYwMDEgNC45NjE1NiAxLjYwMDFaIiBmaWxsPSIjZmZmIi8%2BCjxwYXRoIGQ9Ik00Ljk2MTU2IDEwLjM5OTlIMi4yNDE1NkMxLjg4ODEgMTAuMzk5OSAxLjYwMTU2IDEwLjY4NjQgMS42MDE1NiAxMS4wMzk5VjEzLjc1OTlDMS42MDE1NiAxNC4xMTM0IDEuODg4MSAxNC4zOTk5IDIuMjQxNTYgMTQuMzk5OUg0Ljk2MTU2QzUuMzE1MDIgMTQuMzk5OSA1LjYwMTU2IDE0LjExMzQgNS42MDE1NiAxMy43NTk5VjExLjAzOTlDNS42MDE1NiAxMC42ODY0IDUuMzE1MDIgMTAuMzk5OSA0Ljk2MTU2IDEwLjM5OTlaIiBmaWxsPSIjZmZmIi8%2BCjxwYXRoIGQ9Ik0xMy43NTg0IDEuNjAwMUgxMS4wMzg0QzEwLjY4NSAxLjYwMDEgMTAuMzk4NCAxLjg4NjY0IDEwLjM5ODQgMi4yNDAxVjQuOTYwMUMxMC4zOTg0IDUuMzEzNTYgMTAuNjg1IDUuNjAwMSAxMS4wMzg0IDUuNjAwMUgxMy43NTg0QzE0LjExMTkgNS42MDAxIDE0LjM5ODQgNS4zMTM1NiAxNC4zOTg0IDQuOTYwMVYyLjI0MDFDMTQuMzk4NCAxLjg4NjY0IDE0LjExMTkgMS42MDAxIDEzLjc1ODQgMS42MDAxWiIgZmlsbD0iI2ZmZiIvPgo8cGF0aCBkPSJNNCAxMkwxMiA0TDQgMTJaIiBmaWxsPSIjZmZmIi8%2BCjxwYXRoIGQ9Ik00IDEyTDEyIDQiIHN0cm9rZT0iI2ZmZiIgc3Ryb2tlLXdpZHRoPSIxLjUiIHN0cm9rZS1saW5lY2FwPSJyb3VuZCIvPgo8L3N2Zz4K&logoColor=ffffff" alt="Zread">
21+
</a>
22+
</div>
2123

22-
[![Github star](https://img.shields.io/github/stars/chairc/Integrated-Design-Diffusion-Model.svg)](https://github.com/chairc/Integrated-Design-Diffusion-Model/stargazers)
23-
[![Github fork](https://img.shields.io/github/forks/chairc/Integrated-Design-Diffusion-Model?color=eb6ea5)](https://github.com/chairc/Integrated-Design-Diffusion-Model/forks)
24-
[![Gitee star](https://gitee.com/chairc/Integrated-Design-Diffusion-Model/badge/star.svg?theme=blue)](https://gitee.com/chairc/Integrated-Design-Diffusion-Model)
25-
[![Gitee fork](https://gitee.com/chairc/Integrated-Design-Diffusion-Model/badge/fork.svg?theme=blue)](https://gitee.com/chairc/Integrated-Design-Diffusion-Model)
26-
[![GitCode star](https://gitcode.com/chairc/Integrated-Design-Diffusion-Model/star/badge.svg)](https://gitcode.com/chairc/Integrated-Design-Diffusion-Model)
24+
<div align="center">
25+
<a href="https://github.com/chairc/Integrated-Design-Diffusion-Model/blob/main/LICENSE" target="_blank">
26+
<img src="https://img.shields.io/github/license/chairc/Integrated-Design-Diffusion-Model" alt="License">
27+
</a>
28+
<a href="https://github.com/chairc/Integrated-Design-Diffusion-Model/issues" target="_blank">
29+
<img src="https://img.shields.io/github/issues/chairc/Integrated-Design-Diffusion-Model.svg" alt="GitHub issues">
30+
</a>
31+
<a href="https://github.com/chairc/Integrated-Design-Diffusion-Model/releases" target="_blank">
32+
<img src="https://img.shields.io/github/v/release/chairc/Integrated-Design-Diffusion-Model" alt="GitHub release">
33+
</a>
34+
<a href="#" target="_blank">
35+
<img src="https://img.shields.io/github/downloads/chairc/Integrated-Design-Diffusion-Model/total?color=3eb370" alt="GitHub Downloads">
36+
</a>
37+
</div>
2738

39+
<div align="center">
40+
<a href="https://github.com/chairc/Integrated-Design-Diffusion-Model/stargazers" target="_blank">
41+
<img src="https://img.shields.io/github/stars/chairc/Integrated-Design-Diffusion-Model.svg" alt="GitHub star">
42+
</a>
43+
<a href="https://github.com/chairc/Integrated-Design-Diffusion-Model/forks" target="_blank">
44+
<img src="https://img.shields.io/github/forks/chairc/Integrated-Design-Diffusion-Model?color=eb6ea5" alt="GitHub forks">
45+
</a>
46+
<a href="https://gitee.com/chairc/Integrated-Design-Diffusion-Model" target="_blank">
47+
<img src="https://gitee.com/chairc/Integrated-Design-Diffusion-Model/badge/star.svg?theme=blue" alt="Gitee star">
48+
</a>
49+
<a href="https://gitee.com/chairc/Integrated-Design-Diffusion-Model" target="_blank">
50+
<img src="https://gitee.com/chairc/Integrated-Design-Diffusion-Model/badge/fork.svg?theme=blue" alt="Gitee forks">
51+
</a>
52+
<a href="https://gitcode.com/chairc/Integrated-Design-Diffusion-Model" target="_blank">
53+
<img src="https://gitcode.com/chairc/Integrated-Design-Diffusion-Model/star/badge.svg" alt="GitCode star">
54+
</a>
2855
</div>
2956

3057

@@ -71,7 +98,7 @@ pip install iddm
7198
coloredlogs==15.0.1
7299
gradio>=5.31.0
73100
matplotlib==3.7.1
74-
numpy==1.25.0
101+
numpy>=1.25.0
75102
Pillow==10.3.0
76103
Requests>=2.32.4
77104
scikit-image==0.22.0
@@ -132,6 +159,7 @@ pip install . # 或者 python setup.py install
132159
- [x] [2025-08-01] 增加LDM方法，**支持生成512*512图像**，使用Latent方式降低显存消耗
133160
- [x] [2025-08-21] 支持flash-attn，快速并节省显存
134161
- [x] [2025-10-01] 增加Docker部署文件
162+
- [x] [2025-12-16] 增加自定义日志模组
135163
- [ ] [待定] 重构项目利用百度飞桨框架
136164

137165

‎deploy/deploy_server.py‎

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -23,10 +23,8 @@
2323
import os
2424
import sys
2525
import json
26-
import logging
2726
import uuid
2827

29-
import coloredlogs
3028
import uvicorn
3129

3230
from fastapi import FastAPI
@@ -38,9 +36,9 @@
3836
from iddm.tools.generate import Generator, init_generate_args
3937
from iddm.utils import save_images
4038
from iddm.utils.processing import image_to_base64
39+
from iddm.utils.logger import get_logger
4140

42-
logger = logging.getLogger(__name__)
43-
coloredlogs.install(level="INFO")
41+
logger = get_logger(name=__name__)
4442
app = FastAPI()
4543

4644

‎deploy/deploy_socket.py‎

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,9 +25,7 @@
2525
import json
2626
import uuid
2727
import socket
28-
import logging
2928
import threading
30-
import coloredlogs
3129

3230
from torchvision import transforms
3331

@@ -36,9 +34,9 @@
3634
from iddm.utils.utils import save_images
3735
from iddm.utils.processing import image_to_base64
3836
from iddm.config.version import get_version_banner
37+
from iddm.utils.logger import get_logger
3938

40-
logger = logging.getLogger(__name__)
41-
coloredlogs.install(level="INFO")
39+
logger = get_logger(name=__name__)
4240

4341

4442
def generate(parse_json_data):

‎docs/en-US/01_get_started.md‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ The following packages are required.
2828
coloredlogs==15.0.1
2929
gradio>=5.31.0
3030
matplotlib==3.7.1
31-
numpy==1.25.0
31+
numpy>=1.25.0
3232
Pillow==10.3.0
3333
Requests>=2.32.4
3434
scikit-image==0.22.0

0 commit comments

Comments
 (0)