No module named matting_numpy_unpool for test.py

[protossw512]

I am encountering this issue when trying to perform testing on images. Full log:

Traceback (most recent call last):
  File "test.py", line 5, in <module>
    from matting_numpy_unpool import generate_trimap
ImportError: No module named matting_numpy_unpool

I think it should be:

from matting import generate_trimap

Since we have that module in matting.py.

Also a minor error in README.md:

In usage:

Usage

simply run: python test.py --alpha --rgb sample: python test.py --alpha=./test_data/alpha/1.png --rgb=./test_data/rgb/1.png

The rgb images folder should be "RGB" insead of "rgb", not a big deal though.

[bhack]

Also --trimap_dilation arg seems unused in the code

[Joker316701882]

@protossw512 You are right. Thank you for your correction!

[Joker316701882]

@bhack Thank u! It's fixed now. : )