From 1e729804f61c8627eb257fba8b83f74e04945db7 Mon Sep 17 00:00:00 2001 From: Joseph Redmon Date: Thu, 3 Aug 2017 04:31:04 -0700 Subject: [PATCH] OK GOD DAMN IT I THINK IT'S RIGHT NOW WHATEVER --- examples/detector.py | 2 +- python/darknet.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/detector.py b/examples/detector.py index 10697bab..aa565d61 100644 --- a/examples/detector.py +++ b/examples/detector.py @@ -8,7 +8,7 @@ sys.path.append(os.path.join(os.getcwd(),'python/')) import darknet as dn -net = dn.load_net("cfg/tiny-yolo.cfg", "tiny-yolo.backup", 0) +net = dn.load_net("cfg/tiny-yolo.cfg", "tiny-yolo.weights", 0) meta = dn.load_meta("cfg/coco.data") r = dn.detect(net, meta, "data/dog.jpg") print r diff --git a/python/darknet.py b/python/darknet.py index b2750aea..aadc6347 100644 --- a/python/darknet.py +++ b/python/darknet.py @@ -119,7 +119,7 @@ if __name__ == "__main__": #meta = load_meta("cfg/imagenet1k.data") #r = classify(net, meta, im) #print r[:10] - net = load_net("cfg/tiny-yolo.cfg", "tiny-yolo.backup", 0) + net = load_net("cfg/tiny-yolo.cfg", "tiny-yolo.weights", 0) meta = load_meta("cfg/coco.data") r = detect(net, meta, "data/dog.jpg") print r