1、下载镜像
docker pull registry.cn-hangzhou.aliyuncs.com/modelscope-repo/modelscope:ubuntu20.04-py37-torch1.11.0-tf1.15.5-1.4.3
环境整体依赖的包比较多,自行安装兼容性可能会出现各种问题,建议优先选择官方打包镜像。
2、启动容器
docker run -it 镜像:标签 /bin/bash
3、进入容器创建自己的目录及python文件
- 3.1 到模型库选择感兴趣的模型,这里选择RetinaFace人脸检测关键点模型
- 3.2 copy python示例脚本
frommodelscope.pipelinesimportpipelinefrommodelscope.utils.constantimportTasksretina_face_detection=pipeline(Tasks.face_detection, 'damo/cv_resnet50_face-detection_retinaface') img_path='https://modelscope.oss-cn-beijing.aliyuncs.com/test/images/retina_face_detection.jpg'result=retina_face_detection(img_path) print(f'face detection output: {result}.')
- 3.3 容器中运行python脚本