File tree Expand file tree Collapse file tree
computer_vision_object_detection Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -222,7 +222,7 @@ def YOLOv4(NUM_CLASS, pretrained=False):
222222 network = Model (input_layer , [conv_sbbox , conv_mbbox , conv_lbbox ])
223223
224224 if pretrained :
225- restore_params (network , model_path = 'model/model .npz' )
225+ restore_params (network , model_path = 'model/yolov4_model .npz' )
226226
227227 return network
228228
@@ -236,7 +236,7 @@ def restore_params(network, model_path='models.npz'):
236236 print ("Download the model file, placed in the /model " )
237237 print ("Weights download: " , weights_url ['link' ], "password:" , weights_url ['password' ])
238238
239- txt_path = 'model/yolov4_config .txt'
239+ txt_path = 'model/yolov4_weights_config .txt'
240240 f = open (txt_path , "r" )
241241 line = f .readlines ()
242242 for i in range (len (line )):
Original file line number Diff line number Diff line change @@ -288,7 +288,7 @@ def restore_params(network, model_path='model.npz'):
288288 print ("Download the model file, placed in the /model " )
289289 print ("Weights download: " , weights_url ['link' ], "password:" , weights_url ['password' ])
290290
291- txt_path = 'model/pose_config .txt'
291+ txt_path = 'model/pose_weights_config .txt'
292292 f = open (txt_path , "r" )
293293 line = f .readlines ()
294294 for i in range (len (line )):
@@ -325,7 +325,7 @@ def CGCNN(pretrained=True):
325325 """
326326 if pretrained :
327327 network = cgcnn_inference ()
328- restore_params (network , model_path = 'model/model .npz' )
328+ restore_params (network , model_path = 'model/lcn_model .npz' )
329329 else :
330330 network = cgcnn_train ()
331331 return network
You can’t perform that action at this time.
0 commit comments