update readme

README.md

@@ -1,24 +1,19 @@
-## Installation
-参考Dockerfile，构建运行镜像
-
 ## Quickstart
 
-### 测试程序
-1. 下载模型：https://modelscope.cn/models/AI-ModelScope/stable-diffusion-v1-5
-
-2. 运行测试程序
-
-修改测试程序`test.py`里面的模型路径，直接执行即可
-
+### 构建镜像
 ```bash
-python3 test.py
+docker build -t diffusers:v0.1 .
 ```
 
-### 批量测试程序
-1. 准备输入数据集`dataset.json`，可以参考示例`dataset.json`
-2. 运行测试程序
+### 模型下载
+模型地址：https://modelscope.cn/models/AI-ModelScope/stable-diffusion-v1-5
+并放到目录：`/mnt/contest_ceph/zhanghao/models/stable-diffusion-v1-5`（如更改目录，请修改后面的执行脚本中的模型路径）
+
+### 测试程序
+1. 准备输入数据集，可以参考示例`dataset.json`
+2. 在docker镜像里运行测试程序，会根据`dataset.json`内容，在`output`目录下生成图片文件。
 ```bash
-python3 main.py --model "/mnt/contest_ceph/zhanghao/models/stable-diffusion-v1-5" --json  "dataset.json" --results "results.json" --outdir "output" --device cuda --dtype fp16
+./run_in_docker.sh
 ```
 
 ## 测试结果