From f3ace905d42c9fe75ad70e68a74cc7578a140b8b Mon Sep 17 00:00:00 2001 From: luoyao Date: Fri, 10 Nov 2023 17:20:06 +0800 Subject: [PATCH] update --- conf/model/feature_point/lightglue/lightglue_config.ini | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/conf/model/feature_point/lightglue/lightglue_config.ini b/conf/model/feature_point/lightglue/lightglue_config.ini index 59c144f0..e41ea9eb 100644 --- a/conf/model/feature_point/lightglue/lightglue_config.ini +++ b/conf/model/feature_point/lightglue/lightglue_config.ini @@ -1,5 +1,5 @@ [LIGHTGLUE] -backend_type="onnx" +backend_type="trt" [LIGHTGLUE_ONNX] model_file_path="../weights/feature_point/lightglue/superpoint_lightglue_end2end.onnx" @@ -7,11 +7,12 @@ model_threads_num=4 compute_backend="cuda" gpu_device_id=0 match_thresh=0.5 -long_side_length=1024.0 +long_side_length=512.0 [LIGHTGLUE_TRT] -model_file_path="../weights/feature_point/lightglue/superpoint_lightglue_end2end.engine" +model_file_path="../weights/feature_point/lightglue/superpoint_lightglue_e2d_for_trt.engine" long_side_length=512.0 +max_matched_points=2048 [BACKEND_DICT] onnx=0