jingze_cheng cfb4a14b31 fix: 裁剪bbox以防止超出图像边界 | 8 月之前 | |
---|---|---|
core | 8 月之前 | |
deploy | 11 月之前 | |
minimal_client_server_example | 2 年之前 | |
sx_utils | 1 年之前 | |
templates | 2 年之前 | |
test_imgs | 2 年之前 | |
testing | 2 年之前 | |
.DS_Store | 2 年之前 | |
.gitignore | 8 月之前 | |
Dockerfile | 8 月之前 | |
Dockerfile.base | 1 年之前 | |
Makefile | 1 年之前 | |
README.md | 8 月之前 | |
client.py | 2 年之前 | |
create_md.py | 2 年之前 | |
cutimgs.py | 2 年之前 | |
docker-compose.yml | 2 年之前 | |
environment.yml | 1 年之前 | |
kubeconfig-sxkj | 1 年之前 | |
kubeconfig-sxkjprod | 1 年之前 | |
requirements.txt | 8 月之前 | |
run.py | 1 年之前 | |
server.py | 8 月之前 | |
test.py | 1 年之前 | |
writeMD.py | 2 年之前 |
基于 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,预测部署的代码实现大部分是一致的。