[Model] Optimizer RVM Postprocess (#679)

* add paddle_trt in benchmark

* update benchmark in device

* update benchmark

* update result doc

* fixed for CI

* update python api_docs

* update index.rst

* add runtime cpp examples

* deal with comments

* Update infer_paddle_tensorrt.py

* Add runtime quick start

* deal with comments

* fixed reused_input_tensors&&reused_output_tensors

* fixed docs

* fixed headpose typo

* fixed typo

* refactor yolov5

* update model infer

* refactor pybind for yolov5

* rm origin yolov5

* fixed bugs

* rm cuda preprocess

* fixed bugs

* fixed bugs

* fixed bug

* fixed bug

* fix pybind

* rm useless code

* add convert_and_permute

* fixed bugs

* fixed im_info for bs_predict

* fixed bug

* add bs_predict for yolov5

* Add runtime test and batch eval

* deal with comments

* fixed bug

* update testcase

* fixed batch eval bug

* fixed preprocess bug

* refactor yolov7

* add yolov7 testcase

* rm resize_after_load and add is_scale_up

* fixed bug

* set multi_label true

* optimize rvm preprocess

* optimizer rvm postprocess

* fixed bug

* deal with comments

Co-authored-by: Jason <928090362@qq.com>
Co-authored-by: Jason <jiangjiajun@baidu.com>
This commit is contained in:
WJJ1995
2022-11-25 13:17:09 +08:00
committed by GitHub
parent 124f9f8115
commit 1da8c523b0
6 changed files with 25 additions and 3 deletions

View File

@@ -59,6 +59,13 @@ class RobustVideoMatting(FastDeployModel):
"""
return self._model.video_mode
@property
def swap_rb(self):
"""
Whether convert to RGB, Set to false if you have converted YUV format images to RGB outside the model, dafault true
"""
return self._model.swap_rb
@size.setter
def size(self, wh):
"""
@@ -79,3 +86,12 @@ class RobustVideoMatting(FastDeployModel):
assert isinstance(
value, bool), "The value to set `video_mode` must be type of bool."
self._model.video_mode = value
@swap_rb.setter
def swap_rb(self, value):
"""
Set swap_rb property, the default is true
"""
assert isinstance(
value, bool), "The value to set `swap_rb` must be type of bool."
self._model.swap_rb = value