update

MaybeShewill-CV · Nov 17, 2023 · b05c00a · b05c00a
1 parent c4414fa
commit b05c00a
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/conf/model/scene_segmentation/hrnet/hrnetw48_ccd_fv_segmentation_cfg.ini b/conf/model/scene_segmentation/hrnet/hrnetw48_ccd_fv_segmentation_cfg.ini
@@ -0,0 +1,23 @@
+[HRNET_SEGMENTATION]
+backend_type="trt"
+
+[HRNET_SEGMENTATION_TRT]
+model_file_path="../weights/scene_segmentation/hrnet/hrnetw48_ccd_fp32.engine"
+
+[HRNET_SEGMENTATION_ONNX]
+model_file_path="../weights/scene_segmentation/hrnet/hrnetw48_ccd.onnx"
+model_threads_num=4
+compute_backend="cuda"
+gpu_device_id=0
+
+[HRNET_SEGMENTATION_MNN]
+model_file_path="../weights/scene_segmentation/hrnet/hrnetw48_ccd.mnn"
+model_threads_num=4
+compute_backend="cpu"
+backend_precision_mode=0
+backend_power_mode=0
+
+[BACKEND_DICT]
+trt=0
+onnx=1
+mnn=2