1. 下载并解压源码

解压后路径：Improved_diffusion_module

2. 安装依赖

                    cd Improved_diffusion_module
                    pip install -e .
                    

3. 配置数据集路径

数据集示例路径：
                D:\Python Project\Deep-Traffic\pre-processing\4_Png_16_CTU\Train\Coinminer
                    

4. 启动训练

python scripts/image_train.py \
                        --data_dir "你的数据集路径" \
                        --image_size 16 \
                        --num_channels 128 \
                        --diffusion_steps 50 \
                        --noise_schedule cosine \
                        --lr 1e-4 \
                        --batch_size 16

5. 生成增强数据

python scripts/image_sample.py \
                        --model_path "训练好的模型路径" \
                        --image_size 16 \
                        --diffusion_steps 50