Update: Add parameter explanation.

README.md

@@ -219,14 +219,14 @@ We named this project IDDM: Industrial Defect Diffusion Model. It aims to reprod
 | --------------- | :---------: | ------------------------------- | :--: | ------------------------------------------------------------ |
 | --conditional   |             | Enable conditional generation   | bool | If enabled, allows custom configuration, such as modifying classes or classifier-free guidance interpolation weights |
 | --generate_name |             | File name                       | str  | File name to initialize the model for saving purposes        |
-| --image_size    |             | Input image size                | int  | Size of input images, adaptive input/output size             |
+| --image_size    |             | Input image size                | int  | Size of input images, adaptive input/output size. if class name is -1 and conditional is True, the model would output one image per class. |
 | --num_images    |             | Number of generated images      | int  | Number of images to generate                                 |
 | --weight_path   |             | Path to model weights           | str  | Path to the model weights file, required for network generation |
 | --result_path   |             | Save path                       | str  | Path to save the generated images                            |
 | --sample        |             | Sampling method                 | str  | Set the sampling method type, currently supporting DDPM and DDIM. |
 | --act           |             | Activation function             | str  | Activation function selection. Currently supports gelu, silu, relu, relu6 and lrelu. If you do not set the same activation function as the model, mosaic phenomenon will occur. |
 | --num_classes   |      ✓      | Number of classes               | int  | Number of classes for classification                         |
-| --class_name    |      ✓      | Class name                      | int  | Index of the class to generate images for                    |
+| --class_name    |      ✓      | Class name                      | int  | Index of the class to generate images. if class name is `-1`, the model would output one image per class. |
 | --cfg_scale     |      ✓      | Classifier-free guidance weight | int  | Weight for classifier-free guidance interpolation, for better generation model performance |
 
 ### Result

README_zh.md

@@ -224,14 +224,14 @@
 | --------------- | :------: | -------------------------------- | :------: | ------------------------------------------------------------ |
 | --conditional   |          | 开启条件生成                     |   bool   | 若开启可修改自定义配置，例如修改类别、classifier-free guidance插值权重 |
 | --generate_name |          | 文件名称                         |   str    | 初始化模型的文件名称，用于设置保存信息                       |
-| --image_size    |          | 输入图像大小                     |   int    | 输入图像大小，自适应输入输出尺寸                             |
+| --image_size    |          | 输入图像大小                     |   int    | 输入图像大小，自适应输入输出尺寸。如果输入为-1并且开启条件生成为真，则模型为每类输出一张图片 |
 | --num_images    |          | 生成图片个数                     |   int    | 单次生成图片个数                                             |
 | --weight_path   |          | 权重路径                         |   str    | 模型权重路径，网络生成需要加载文件                           |
 | --result_path   |          | 保存路径                         |   str    | 保存路径                                                     |
 | --sample        |          | 采样方式                         |   str    | 设置采样器类别，当前支持ddpm，ddim                           |
 | --act           |          | 激活函数                         |   str    | 激活函数选择，目前支持gelu、silu、relu、relu6和lrelu。如果不选择，会产生马赛克现象 |
 | --num_classes   |    是    | 类别个数                         |   int    | 类别个数，用于区分类别                                       |
-| --class_name    |    是    | 类别名称                         |   int    | 类别序号，用于对指定类别生成                                 |
+| --class_name    |    是    | 类别名称                         |   int    | 类别序号，用于对指定类别生成。如果输入为-1，则模型为每类输出一张图片 |
 | --cfg_scale     |    是    | classifier-free guidance插值权重 |   int    | classifier-free guidance插值权重，用户更好生成模型效果       |