jingze_cheng cfb4a14b31 fix: 裁剪bbox以防止超出图像边界 | пре 7 месеци | |
---|---|---|
core | пре 7 месеци | |
deploy | пре 10 месеци | |
minimal_client_server_example | пре 2 година | |
sx_utils | пре 1 година | |
templates | пре 2 година | |
test_imgs | пре 2 година | |
testing | пре 2 година | |
.DS_Store | пре 2 година | |
.gitignore | пре 8 месеци | |
Dockerfile | пре 7 месеци | |
Dockerfile.base | пре 1 година | |
Makefile | пре 1 година | |
README.md | пре 7 месеци | |
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,预测部署的代码实现大部分是一致的。