mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
OK ADDED ONE MORE EXAMPLE BUT NOW I'M REALLY GONE FOR SERIOUS BACK IN A WHILE
This commit is contained in:
parent
b34082019d
commit
8327154a67
25
examples/detector.py
Normal file
25
examples/detector.py
Normal file
@ -0,0 +1,25 @@
|
||||
# Stupid python path shit.
|
||||
# Instead just add darknet.py to somewhere in your python path
|
||||
# OK actually that might not be a great idea, idk, work in progress
|
||||
# Use at your own risk. or don't, i don't care
|
||||
|
||||
import sys, os
|
||||
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)
|
||||
meta = dn.load_meta("cfg/coco.data")
|
||||
r = dn.detect(net, meta, "data/dog.jpg")
|
||||
print r
|
||||
|
||||
# And then down here you could detect a lot more images like:
|
||||
r = dn.detect(net, meta, "data/eagle.jpg")
|
||||
print r
|
||||
r = dn.detect(net, meta, "data/giraffe.jpg")
|
||||
print r
|
||||
r = dn.detect(net, meta, "data/horses.jpg")
|
||||
print r
|
||||
r = dn.detect(net, meta, "data/person.jpg")
|
||||
print r
|
||||
|
Loading…
Reference in New Issue
Block a user