mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-07 09:31:35 +08:00
a8458e6729
* 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 Co-authored-by: Jason <jiangjiajun@baidu.com>
37 lines
1.0 KiB
Python
37 lines
1.0 KiB
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(model_url, ".", show_progress=True)
|
|
fd.download(test_jpg_url, ".", show_progress=True)
|
|
|
|
try:
|
|
tar = tarfile.open("unet_Cityscapes.tar.gz", "r:gz")
|
|
file_names = tar.getnames()
|
|
for file_name in file_names:
|
|
tar.extract(file_name, ".")
|
|
tar.close()
|
|
except Exception as e:
|
|
raise Exception(e)
|
|
|
|
# 加载模型
|
|
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)
|