mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
37 lines
1014 B
Python
37 lines
1014 B
Python
# 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
|
|
import pdb
|
|
|
|
<<<<<<< HEAD
|
|
net = dn.load_net("cfg/yolo-tag.cfg", "yolo-tag_final.weights", 0)
|
|
meta = dn.load_meta("cfg/openimages.data")
|
|
pdb.set_trace()
|
|
rr = dn.detect(net, meta, 'data/dog.jpg')
|
|
print rr
|
|
pdb.set_trace()
|
|
=======
|
|
dn.set_gpu(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
|
|
>>>>>>> 16686cec576580489ab3c7c78183e6efeafae780
|
|
|
|
# And then down here you could detect a lot more images like:
|
|
rr = dn.detect(net, meta, "data/eagle.jpg")
|
|
print rr
|
|
rr = dn.detect(net, meta, "data/giraffe.jpg")
|
|
print rr
|
|
rr = dn.detect(net, meta, "data/horses.jpg")
|
|
print rr
|
|
rr = dn.detect(net, meta, "data/person.jpg")
|
|
print rr
|
|
|