jingze_cheng db1e83b48c fix: 修复依赖版本冲突 | 7 months ago | |
---|---|---|
core | 7 months ago | |
deploy | 10 months ago | |
minimal_client_server_example | 2 years ago | |
sx_utils | 1 year ago | |
templates | 7 months ago | |
test_imgs | 2 years ago | |
testing | 2 years ago | |
.DS_Store | 2 years ago | |
.gitignore | 8 months ago | |
Dockerfile | 7 months ago | |
Dockerfile.base | 1 year ago | |
Makefile | 1 year ago | |
README.md | 7 months ago | |
client.py | 2 years ago | |
create_md.py | 2 years ago | |
cutimgs.py | 2 years ago | |
docker-compose.yml | 2 years ago | |
environment.yml | 7 months ago | |
kubeconfig-sxkj | 1 year ago | |
kubeconfig-sxkjprod | 1 year ago | |
requirements.txt | 7 months ago | |
run.py | 1 year ago | |
server.py | 7 months ago | |
test.py | 1 year ago | |
writeMD.py | 2 years ago |
基于 Official YOLOv7 和 PaddleYOLO,用于水平框版面分析。
conda env create -f environment.yml
python run.py --port 8080
make all
模型名称 | 模型配置 |
---|---|
Official YOLOv7 | ./core/detectors/yolov7.py |
PaddleYOLO YOLOv7P6-E6 | ./core/detectors/paddle_yolo/__init__.py |
如果更新了模型权重,请同时修改创建镜像时的下载地址,如:
$ cat Dockerfile
...
# 模型权重下载地址
RUN wget ftp://192.168.199.31/yolov7p6_e6_300e/20240314/yolov7p6_e6_300e_coco.zip
...
版面分析服务的简化流程为:
图片 + 模型名称
|
| (根据模型名称,对图片进行相应的布局检测)
v
布局检测框集
其中基于 PaddleYOLO 的布局检测实现中还包含图像预处理,模型预测,后处理等。源码:PaddleYOLO/deploy/python/infer.py#L62
注:PaddleYOLO 基于 PaddleDetection,预测部署的代码实现大部分是一致的。