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))