From 7656d315d4402b97c7237e206e8d96e4d623c633 Mon Sep 17 00:00:00 2001
From: Shrishti Hore <40817931+ShrishtiHore@users.noreply.github.com>
Date: Wed, 1 Jan 2020 13:44:47 +0530
Subject: [PATCH] Update darknet.py

---
 darkflow/dark/darknet.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/darkflow/dark/darknet.py b/darkflow/dark/darknet.py
index 841245610..e65c0abef 100644
--- a/darkflow/dark/darknet.py
+++ b/darkflow/dark/darknet.py
@@ -47,7 +47,7 @@ def get_weight_src(self, FLAGS):
             '{} not found'.format(FLAGS.load)
             self.src_bin = FLAGS.load
             name = loader.model_name(FLAGS.load)
-            cfg_path = os.path.join(FLAGS.config, name + '.cfg')
+            cfg_path = os.path.join(FLAGS.config)
             if not os.path.isfile(cfg_path):
                 warnings.warn(
                     '{} not found, use {} instead'.format(
@@ -83,4 +83,4 @@ def load_weights(self):
         for layer in self.layers: layer.load(wgts_loader)
         
         stop = time.time()
-        print('Finished in {}s'.format(stop - start))
\ No newline at end of file
+        print('Finished in {}s'.format(stop - start))