digitalbrain79 / pyyolo

Simple python wrapper for YOLO.
126 stars 65 forks source link

Cuda malloc failed #11

Closed hungpt297 closed 7 years ago

hungpt297 commented 7 years ago

When running example.py got error: layer filters size input output 0 Cuda malloc failed : File exists python: ./darknet/src/utils.c:216: error: Assertion `0' failed. Aborted (core dumped)

digitalbrain79 commented 7 years ago

I think it is related to darknet. Could you run darknet only without pyyolo?

hungpt297 commented 7 years ago

Yes, it is problem in darknet, Example run with config datacfg = 'cfg/voc.data' cfgfile = 'cfg/yolo.cfg' weightfile = '../yolo.weights'

Many thanks.