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:

2
examples/nightmare.c
View File

@ -62,7 +62,7 @@ void optimize_picture(network *net, image orig, int max_layer, float scale, floa
cuda_free(net->delta_gpu);
net->delta_gpu = 0;
#else
net->input = im.data;
copy_cpu(net->inputs, im.data, 1, net->input, 1);
net->delta = delta.data;
forward_network(net);
copy_cpu(last.outputs, last.output, 1, last.delta, 1);