fixed readme

.idea/workspace.xml

@@ -4,11 +4,10 @@
     <option name="autoReloadType" value="SELECTIVE" />
   </component>
   <component name="ChangeListManager">
-    <list default="true" id="01d3e030-fd3e-4982-8660-10d478b2149f" name="Default Changelist" comment="fixed readme">
-      <change afterPath="$PROJECT_DIR$/inference.py" afterDir="false" />
+    <list default="true" id="01d3e030-fd3e-4982-8660-10d478b2149f" name="Default Changelist" comment="fixed readme and add inference api">
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/README.md" beforeDir="false" afterPath="$PROJECT_DIR$/README.md" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/requirement.txt" beforeDir="false" afterPath="$PROJECT_DIR$/requirement.txt" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/model.properties" beforeDir="false" afterPath="$PROJECT_DIR$/model.properties" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -37,19 +36,20 @@
     <option name="hideEmptyMiddlePackages" value="true" />
     <option name="showLibraryContents" value="true" />
   </component>
-  <component name="PropertiesComponent">{
-  &quot;keyToString&quot;: {
-    &quot;WebServerToolWindowFactoryState&quot;: &quot;false&quot;,
-    &quot;git-widget-placeholder&quot;: &quot;Rebasing main&quot;,
-    &quot;last_opened_file_path&quot;: &quot;D:/sugon/work_space/git/model_zoo&quot;,
-    &quot;node.js.detected.package.eslint&quot;: &quot;true&quot;,
-    &quot;node.js.detected.package.tslint&quot;: &quot;true&quot;,
-    &quot;node.js.selected.package.eslint&quot;: &quot;(autodetect)&quot;,
-    &quot;node.js.selected.package.tslint&quot;: &quot;(autodetect)&quot;,
-    &quot;nodejs_package_manager_path&quot;: &quot;npm&quot;,
-    &quot;vue.rearranger.settings.migration&quot;: &quot;true&quot;
+  <component name="PropertiesComponent"><![CDATA[{
+  "keyToString": {
+    "WebServerToolWindowFactoryState": "false",
+    "git-widget-placeholder": "main",
+    "last_opened_file_path": "D:/sugon/work_space/git/model_zoo",
+    "node.js.detected.package.eslint": "true",
+    "node.js.detected.package.tslint": "true",
+    "node.js.selected.package.eslint": "(autodetect)",
+    "node.js.selected.package.tslint": "(autodetect)",
+    "nodejs_package_manager_path": "npm",
+    "settings.editor.selected.configurable": "File.Encoding",
+    "vue.rearranger.settings.migration": "true"
   }
-}</component>
+}]]></component>
   <component name="SpellCheckerSettings" RuntimeDictionaries="0" Folders="0" CustomDictionaries="0" DefaultDictionary="application-level" UseSingleDictionary="true" transferred="true" />
   <component name="TaskManager">
     <task active="true" id="Default" summary="Default task">
@@ -60,8 +60,17 @@
       <updated>1694486720086</updated>
       <workItem from="1698909897462" duration="1290000" />
       <workItem from="1699238067729" duration="75000" />
-      <workItem from="1700186034106" duration="5752000" />
+      <workItem from="1700186034106" duration="8015000" />
+    </task>
+    <task id="LOCAL-00001" summary="fixed readme and add inference api">
+      <option name="closed" value="true" />
+      <created>1700192449643</created>
+      <option name="number" value="00001" />
+      <option name="presentableId" value="LOCAL-00001" />
+      <option name="project" value="LOCAL" />
+      <updated>1700192449643</updated>
     </task>
+    <option name="localTasksCounter" value="2" />
     <servers />
   </component>
   <component name="TypeScriptGeneratedFilesManager">
@@ -80,6 +89,7 @@
   </component>
   <component name="VcsManagerConfiguration">
     <MESSAGE value="fixed readme" />
-    <option name="LAST_COMMIT_MESSAGE" value="fixed readme" />
+    <MESSAGE value="fixed readme and add inference api" />
+    <option name="LAST_COMMIT_MESSAGE" value="fixed readme and add inference api" />
   </component>
 </project>
\ No newline at end of file

README.md

@@ -24,6 +24,8 @@ stable diffusion 对文本进行处理并进行编码生成文本向量，对初
 ```
 docker pull image.sourcefind.cn:5000/dcu/admin/base/paddlepaddle:2.4.2-centos7.6-dtk-23.04-py39-latest 
 docker run --shm-size 16g --network=host --name=image_generate_paddle --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/imagegenerate_paddle:/home/imagegenerate_paddle -it <Your Image ID> bash
+cd imagegenerate_paddle
+pip install -r requirements.txt
 ```
 ### Dockerfile（方法二）
 此处提供dockerfile的使用方法
@@ -39,7 +41,7 @@ docker run --shm-size 16g --network=host --name=image_generate_paddle --privileg
 关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。
 ```
 DTK驱动：dtk23.04
-python：python3.8
+python：python3.9
 paddle:2.4.2
 ```
 `Tips：以上dtk驱动、python、paddle等DCU相关工具版本需要严格一一对应`
@@ -66,7 +68,7 @@ pip install -r requirements.txt
 - 方法二：使用notebook运行main.ipynb
   - 根据notebook提示运行代码
 ## 测试
-- 测试环境、功能是否够能够正常
+#### 测试环境、功能是否够能够正常
 ```python
 python varify.py 
 ```
@@ -77,12 +79,13 @@ python varify.py
 |:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------:|:------------------------------------------:|:------------------------------------------------------------------------------------------------------------------------------:|
 |'prompt':'extremely detailed CG unity 8k wallpaper,black long hair,cute face,1 adult girl,happy, green skirt dress, flower pattern in dress,solo,green gown,art of light novel,in field'<br>"negative_prompt":'lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry' |         ![图生图](./resources/cat2.jpg)          | "prompt": '<Alice> at the lake',<br>"negative_prompt": ''<br>![图生图](./resources/Alices/1666360027.7809527_SEED_3534839969.png) |
 |                                                                                                                                                                                     ![文生图](./resources/pic_readme/txt2img.png)                                                                                                                                                                                      | ![图生图](./resources/pic_readme/img2img.png) |                                           ![定制化训练](./resources/pic_readme/train.png)                                           |
-
+## 精度
+无
 ## 应用场景
 ### 算法类别
-`NLP`
+`文生图、图生图、文生图定制化训练`
 ### 热点应用行业
-`广媒,设计,其他`
+`广媒,设计`
 ## 源码仓库及问题反馈
 - https://developer.hpccube.com/codes/modelzoo/imagegenerate_paddle
 ## 参考资料

docker/Dockerfile

 FROM image.sourcefind.cn:5000/dcu/admin/base/paddlepaddle:2.4.2-centos7.6-dtk-23.04-py39-latest
 RUN source /opt/dtk/env.sh
-COPY ../requirments.txt requirments.txt
+COPY requirments.txt requirments.txt
 RUN pip3 install -r requirements.txt
 

model.properties

 # 模型唯一标识
 modelCode = 304
 # 模型名称
-modelName=image_generate_paddle
+modelName=ImageGenerate_paddle
 # 模型描述
 modelDescription=文生图、图生图模型是一种基于深度学习的图像生成算法，可应用于艺术创作提供创作思路
 # 应用场景
-appScenario=推理,训练,NLP,设计,广媒
+appScenario=推理,训练,文生图,图生图,文生图定制化训练,设计,广媒
 # 框架类型
 frameType=paddle
\ No newline at end of file