Merge pull request chairc#134 from chairc/dev

docs: Update running locally in README.

Author: chairc

README.md

@@ -51,13 +51,21 @@ We named this project IDDM: Integrated Design Diffusion Model. It aims to reprod
 
 If you have any questions, please check [**the existing issues**](https://github.com/chairc/Integrated-Design-Diffusion-Model/issues/9) first. If the issue persists, feel free to open a new one for assistance, or you can contact me via email at chenyu1998424@gmail.com or chairc1998@163.com. **If you think my project is interesting, please give me a ⭐⭐⭐Star⭐⭐⭐ :)**
 
-#### Running  Locally
+#### Running Locally
 
 Use the `git clone` or directly download the `zip` file of this repository's code, and then configure the environment locally to run it.
 
 ```bash
 git clone https://github.com/chairc/Integrated-Design-Diffusion-Model.git
 cd Integrated-Design-Diffusion-Model
+# Run the project in a virtual environment (recommended)
+conda create -n iddm python=3.10
+pip install -r requirements.txt
+# Train the model
+cd iddm/tools
+python train.py --xxx xxx # Replace --xxx with your training parameters
+# Generate images
+python generate.py --xxx xxx # Replace --xxx with your generation parameters
 ```
 
 #### Installation
@@ -114,7 +122,7 @@ pip install . # Or python setup.py install
 - [x] [2023-09-16] Support other image generation.
 - [x] [2023-11-09] Adding a more advanced U-Net network model.
 - [x] [2023-11-09] Support generate larger-sized images.
-- [x] [2023-12-06] Reconstruct the overall structure of the model.
+- [x] [2023-12-06] Refactor the overall structure of the model.
 - [x] [2024-01-23] Adding visual webui training interface.
 - [x] [2024-02-18] Support low-resolution generated images for super-resolution enhancement.[~~Super resolution model, the effect is uncertain~~]
 - [x] [2024-03-12] Adding PLMS Sampling Method.
@@ -124,10 +132,10 @@ pip install . # Or python setup.py install
 - [x] [2024-11-13] Adding the deployment of image-generating Sockets and Web server.
 - [x] [2024-11-26] Adding PSNR and SSIM calculators to verify super resolution image quality.
 - [x] [2024-12-10] Adding pretrain model download.
-- [x] [2024-12-25] Reconstruct the overall structure of the trainer.
+- [x] [2024-12-25] Refactor the overall structure of the trainer.
 - [x] [2025-03-08] Support PyPI install.
 - [ ] [Maybe 2025-01-31] Adding the deployment of Docker and image.
-- [ ] [To be determined] Reconstruct the project by Baidu PaddlePaddle.
+- [ ] [To be determined] Refactor the project by Baidu PaddlePaddle.
 - [ ] [To be determined] ~~Use Latent Diffusion and reduce GPU memory usage~~
 
 

README_zh.md

@@ -57,6 +57,14 @@
 ```bash
 git clone https://github.com/chairc/Integrated-Design-Diffusion-Model.git
 cd Integrated-Design-Diffusion-Model
+# 在虚拟环境中运行项目（推荐）
+conda create -n iddm python=3.10
+pip install -r requirements.txt
+# 训练模型
+cd iddm/tools
+python train.py --xxx xxx # 用个人参数替换-xxx
+# 生成图像
+python generate.py --xxx xxx # 用个人参数替换-xxx
 ```
 
 #### 安装