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