thtrieu / darkflow

Translate darknet to tensorflow. Load trained weights, retrain/fine-tune using tensorflow, export constant graph def to mobile devices
GNU General Public License v3.0
6.13k stars 2.08k forks source link

Update darknet.py #1116

Open ShrishtiHore opened 4 years ago

ShrishtiHore commented 4 years ago

Due to this issue : Parsing yolov2.cfg /content/darkflow/darkflow/dark/darknet.py:54: UserWarning: ./cfg/yolov2.cfg not found, use yolov2.cfg instead cfg_path, FLAGS.model))

KeyError Traceback (most recent call last)

in () 4 'threshold': 0.3 5 } ----> 6 tfnet = TFNet(options) 4 frames /content/darkflow/darkflow/utils/process.py in parser(model) 27 if '[' in line: 28 if layer != dict(): ---> 29 if layer['type'] == '[net]': 30 h = layer['height'] 31 w = layer['width'] KeyError: 'type'