search

MarvinTeichmann / KittiBox

A car detection model implemented in Tensorflow.
MIT License
401 stars 169 forks source link

ModuleNotFoundError: No module named 'stitch_wrapper' #34

Open anthony-chaudhary opened 7 years ago

anthony-chaudhary commented 7 years ago

Hello :)

This is a really cool idea and implementation!

I bumped into what appears to be a small bug with python 3.6 on Ubuntu 16.04.

While running python evaluate.py this error is returned: ModuleNotFoundError: No module named 'stitch_wrapper'

The fix below works for me. Not sure if it's a general issue or not. Figured I would post here in case others run into it:

  1. run cd submodules/utils && make as directed in readme, if you haven't already
  2. rename stitch_wrapper.pyx to stitch_wrapper
  3. change the import statement on line 67 in train_utils to add utils.
    ie -> from utils.stitch_wrapper import stitch_rects
MarvinTeichmann commented 7 years ago

Seams to be a python 3.6 / 2.7 compatibility error. If there is a way to make it work under both versions, feel free to create a pull request.