mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

I WISH I HAD SOME TESTS THOUGH

Browse Source
This commit is contained in:
Joseph Redmon
2017-11-21 11:34:46 -08:00
parent 16686cec57
commit 56be49aa48
6 changed files with 16 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
examples/detector.py
View File

@ -7,11 +7,12 @@ import sys, os
sys.path.append(os.path.join(os.getcwd(),'python/'))
import darknet as dn
import pdb
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")
net = dn.load_net("cfg/yolo-thor.cfg", "/home/pjreddie/backup/yolo-thor_final.weights", 0)
meta = dn.load_meta("cfg/thor.data")
r = dn.detect(net, meta, "data/bedroom.jpg")
print r
# And then down here you could detect a lot more images like: