scale predict box back to image space

experiencor / keras-yolo2

Easy training on custom dataset. Various backends (MobileNet and SqueezeNet) supported. A YOLO demo to detect raccoon run entirely in brower is accessible at https://git.io/vF7vI (not on Windows).

MIT License

1.73k stars 784 forks source link

scale predict box back to image space #261

Closed pure-water closed 6 years ago

pure-water commented 6 years ago

Scale predict box back to image space

sroj commented 6 years ago

Hi. Getting this error on this PR:

Traceback (most recent call last):
  File "train.py", line 104, in <module>
    _main_(args)
  File "train.py", line 100, in _main_
    debug              = config['train']['debug'])
  File ".../frontend.py", line 341, in train
    average_precisions = self.evaluate(valid_generator)
  File ".../frontend.py", line 373, in evaluate
    raw_height, raw_width, raw_channels = raw_img.shape
NameError: global name 'raw_img' is not defined

pure-water commented 6 years ago

typo fixed. Try again