Skip to content

Commit 8d9ead0

Browse files
committed
weights rename
1 parent e30d1b0 commit 8d9ead0

2 files changed

Lines changed: 4 additions & 4 deletions

File tree

  • tensorlayer/app

tensorlayer/app/computer_vision_object_detection/yolov4.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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)):

tensorlayer/app/human_pose_estimation/LCN.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)