mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-31 20:02:53 +08:00
0e0dfd9565
* Support new model PaddleSeg * Fix conflict * PaddleSeg add visulization function * fix bug * Fix BindPPSeg wrong name * Fix variable name * Update by comments * Add ppseg-unet example python version * Change the way to decompress model file * Visualize resize mask back to original image size * Update paddleseg support * Add essential files to support ppseg * Support logits matrix resize * Support mask matrix resize * Fix some bugs * Format code * Add code comment * Format code Co-authored-by: Jason <jiangjiajun@baidu.com>
27 lines
837 B
Python
27 lines
837 B
Python
import fastdeploy as fd
|
|
import cv2
|
|
import tarfile
|
|
|
|
# 下载模型和测试图片
|
|
model_url = "https://github.com/felixhjh/Fastdeploy-Models/raw/main/unet_Cityscapes.tar.gz"
|
|
test_jpg_url = "https://paddleseg.bj.bcebos.com/dygraph/demo/cityscapes_demo.png"
|
|
fd.download_and_decompress(model_url, ".")
|
|
fd.download(test_jpg_url, ".", show_progress=True)
|
|
# 加载模型
|
|
model = fd.vision.ppseg.Model("./unet_Cityscapes/model.pdmodel",
|
|
"./unet_Cityscapes/model.pdiparams",
|
|
"./unet_Cityscapes/deploy.yaml")
|
|
|
|
# 预测图片
|
|
im = cv2.imread("./cityscapes_demo.png")
|
|
result = model.predict(im)
|
|
|
|
vis_im = im.copy()
|
|
# 可视化结果
|
|
fd.vision.visualize.vis_segmentation(im, result, vis_im)
|
|
cv2.imwrite("vis_img.png", vis_im)
|
|
|
|
# 输出预测结果
|
|
print(result)
|
|
print(model.runtime_option)
|