ModuleNotFoundError: No module named 'stitch_wrapper'

[anthony-chaudhary]

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]

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.