This commit is contained in:
Daniel Gordon 2018-03-15 13:33:41 -07:00
parent ffa8c699a8
commit 6fc177fa5c

View File

@ -13,7 +13,7 @@ def sample(probs):
return len(probs)-1
def c_array(ctype, values):
arr = (ctype * len(values))()
arr = (ctype*len(values))()
arr[:] = values
return arr
@ -153,4 +153,6 @@ if __name__ == "__main__":
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
print(r)