From e118b7030b02ab5b79b7e75e7a4f7d7f9b3a52e4 Mon Sep 17 00:00:00 2001 From: Chris Date: Tue, 12 Sep 2017 10:14:56 -0600 Subject: [PATCH] bug fix --- test_pretrained.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test_pretrained.py b/test_pretrained.py index 9c30121..04d7993 100644 --- a/test_pretrained.py +++ b/test_pretrained.py @@ -9,7 +9,7 @@ import cv2 from process_pixel_labels import post_process NET_FILE = './models/cbad_train_val.prototxt' -WEIGHT_FILE = './models/cbad_weights.prototxt' +WEIGHT_FILE = './models/cbad_weights.caffemodel' def safe_mkdir(_dir):