@@ -74,7 +74,7 @@ def parse_model_grpc(model_metadata, model_config):
7474
7575class Arcface :
7676
77- def __init__ (self , rec_name = 'arcface_r100_v1' , model_version = '1' , triton_uri = '10.0.15.109:8002 ' , ** kwargs ):
77+ def __init__ (self , rec_name = 'arcface_r100_v1' , model_version = '1' , triton_uri = 'localhost:8001 ' , ** kwargs ):
7878 self .model_name = rec_name
7979 self .model_version = model_version
8080 self .url = triton_uri
@@ -135,7 +135,7 @@ def get_embedding(self,face_img):
135135
136136
137137class Cosface :
138- def __init__ (self , rec_name = 'arcface_r100_v1' , triton_uri = '10.0.15.109:8002 ' , model_version = '1' ,** kwargs ):
138+ def __init__ (self , rec_name = 'arcface_r100_v1' , triton_uri = 'localhost:8001 ' , model_version = '1' ,** kwargs ):
139139 self .model_name = rec_name
140140 self .model_version = model_version
141141 self .url = triton_uri
@@ -196,7 +196,7 @@ def get_embedding(self, face_img):
196196
197197class DetectorInfer :
198198
199- def __init__ (self , model = 'retinaface_r50_v1' , output_order = None , triton_uri = '10.0.15.109:8002 ' , model_version = '1' ,** kwargs ):
199+ def __init__ (self , model = 'retinaface_r50_v1' , output_order = None , triton_uri = 'localhost:8001 ' , model_version = '1' ,** kwargs ):
200200 self .model_name = model
201201 self .model_version = model_version
202202 self .url = triton_uri
0 commit comments