apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-07 01:22:59 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Model] Add Paddle3D smoke model (#1766)

Browse Source 
* add smoke model

* add 3d vis

* update code

* update doc

* mv paddle3d from detection to perception

* update result for velocity

* update code for CI

* add set input data for TRT backend

* add serving support for smoke model

* update code

* update code

* update code

---------

Co-authored-by: DefTruth <31974251+DefTruth@users.noreply.github.com>
This commit is contained in:
yeliang2258
2023-04-14 16:30:56 +08:00
committed by GitHub
parent 81fbd54c9d
commit a509dd8ec1
53 changed files with 2610 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
python/fastdeploy/vision/utils.py Normal file → Executable file
View File

@@ -38,6 +38,19 @@ def detection_to_json(result):
return json.dumps(r_json)
def perception_to_json(result):
r_json = {
"scores": result.scores,
"label_ids": result.label_ids,
"boxes": result.boxes,
"center": result.center,
"observation_angle": result.observation_angle,
"yaw_angle": result.yaw_angle,
"velocity": result.velocity
}
return json.dumps(r_json)
def classify_to_json(result):
r_json = {
"label_ids": result.label_ids,
@@ -151,6 +164,8 @@ def fd_result_to_json(result):
return matting_to_json(result)
elif isinstance(result, C.vision.HeadPoseResult):
return head_pose_to_json(result)
elif isinstance(result, C.vision.PerceptionResult):
return perception_to_json(result)
else:
assert False, "{} Conversion to JSON format is not supported".format(
type(result))
@@ -177,6 +192,18 @@ def json_to_detection(result):
return det_result
def json_to_perception(result):
perception_result = C.vision.PerceptionResult()
perception_result.scores = result['scores']
perception_result.label_ids = result['label_ids']
perception_result.boxes = result['boxes']
perception_result.center = result['center']
perception_result.observation_angle = result['observation_angle']
perception_result.yaw_angle = result['yaw_angle']
perception_result.velocity = result['velocity']
return perception_result
def json_to_classify(result):
cls_result = C.vision.ClassifyResult()
cls_result.label_ids = result['label_ids']