inital commit

Dockerfile

+FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.10.0-centos7.6-dtk-23.04-py37-latest
+COPY requirements.txt requirements.txt
+RUN source /opt/dtk-22.10/env.sh
+RUN cp /usr/share/zoneinfo/Asia/Shanghai /etc/localtime && echo 'Asia/Shanghai' >/etc/timezone 
+ENV LANG C.UTF-8
+RUN pip install -r requirements.txt -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com

README.md

@@ -104,7 +104,7 @@ python predict.py -m model_path -i image.jpg -o output.jpg
 `图像分割`
 
 ### 热点应用行业
-`医疗`
+`医疗`,`交通`,`家居`
 
 ## 源码仓库及问题反馈
 - https://developer.hpccube.com/codes/modelzoo/unet-pytorch

model.properties

@@ -5,6 +5,6 @@ modelName=unet_pytorch
 # 模型描述
 modelDescription=UNet的主要作用是从输入图像中生成一个具有相同空间分辨率的输出图像，其中每个像素被分配到其对应的类别。与其他神经网络架构相比，UNet在图像分割任务中表现出色
 # 应用场景
-appScenario=推理,训练,图像分割,医疗
+appScenario=推理,训练,图像分割,医疗,交通,家居
 # 框架类型
 frameType=pytorch