apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-31 20:02:53 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Update PPSeg Evaluation (#194)

Browse Source 
Update segmentation evaluation

Co-authored-by: Jason <jiangjiajun@baidu.com>
This commit is contained in:
huangjianhui
2022-09-06 16:20:23 +08:00
committed by GitHub
parent 401f3349f8
commit f04dfb39a1
2 changed files with 3 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
fastdeploy/vision/evaluation/__init__.py
View File

@@ -14,3 +14,4 @@
from __future__ import absolute_import from __future__ import absolute_import
from .classify import eval_classify from .classify import eval_classify
from .detection import eval_detection from .detection import eval_detection
from .segmentation import eval_segmentation

21
fastdeploy/vision/evaluation/segmentation.py
View File

@@ -22,8 +22,8 @@ import time
def eval_segmentation(model, data_dir): def eval_segmentation(model, data_dir):
import cv2 import cv2
from utils import Cityscapes from .utils import Cityscapes
from utils import f1_score, calculate_area, mean_iou, accuracy, kappa from .utils import f1_score, calculate_area, mean_iou, accuracy, kappa
assert os.path.isdir( assert os.path.isdir(
data_dir), "The image_file_path:{} is not a directory.".format( data_dir), "The image_file_path:{} is not a directory.".format(
data_dir) data_dir)
@@ -76,20 +76,3 @@ def eval_segmentation(model, data_dir):
average_inference_time average_inference_time
])) ]))
return eval_metrics return eval_metrics
import fastdeploy as fd
#model = fd.vision.segmentation.PaddleSegModel("/huangjianhui/temp/FastDeploy/model_zoo/vision/ppseg/unet/unet_Cityscapes/model.pdmodel",
# "/huangjianhui/temp/FastDeploy/model_zoo/vision/ppseg/unet/unet_Cityscapes//model.pdiparams",
# "/huangjianhui/temp/FastDeploy/model_zoo/vision/ppseg/unet/unet_Cityscapes/deploy.yaml")
#
option = fd.RuntimeOption()
option.use_paddle_backend()
option.use_gpu(3)
model = fd.vision.segmentation.PaddleSegModel(
"/huangjianhui/temp/FastDeploy/model_zoo/vision/ppseg/PP-LiteSeg/output_no_static_size/model.pdmodel",
"/huangjianhui/temp/FastDeploy/model_zoo/vision/ppseg/PP-LiteSeg/output_no_static_size/model.pdiparams",
"/huangjianhui/temp/FastDeploy/model_zoo/vision/ppseg/PP-LiteSeg/output_no_static_size/deploy.yaml",
option)
result = eval_segmentation(model, "/huangjianhui/PaddleSeg/data/cityscapes/")