jingze_cheng db1e83b48c fix: 修复依赖版本冲突 | il y a 7 mois | |
---|---|---|
core | il y a 7 mois | |
deploy | il y a 10 mois | |
minimal_client_server_example | il y a 2 ans | |
sx_utils | il y a 1 an | |
templates | il y a 7 mois | |
test_imgs | il y a 2 ans | |
testing | il y a 2 ans | |
.DS_Store | il y a 2 ans | |
.gitignore | il y a 8 mois | |
Dockerfile | il y a 7 mois | |
Dockerfile.base | il y a 1 an | |
Makefile | il y a 1 an | |
README.md | il y a 7 mois | |
client.py | il y a 2 ans | |
create_md.py | il y a 2 ans | |
cutimgs.py | il y a 2 ans | |
docker-compose.yml | il y a 2 ans | |
environment.yml | il y a 7 mois | |
kubeconfig-sxkj | il y a 1 an | |
kubeconfig-sxkjprod | il y a 1 an | |
requirements.txt | il y a 7 mois | |
run.py | il y a 1 an | |
server.py | il y a 7 mois | |
test.py | il y a 1 an | |
writeMD.py | il y a 2 ans |
基于 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,预测部署的代码实现大部分是一致的。