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