thug-memes
thug-memes copied to clipboard
Add handling for cases in which sub image does not fit on top of base image
trafficstars
For example, if cigar/glasses image does not fit on top of the original image, the traceback is something like this:
File "/Users/user/.virtualenvs/user-venv/lib/python3.5/site-packages/thug/meme/thug.py", line 122, in _draw_on_top background = cv2.bitwise_and(background, background, mask=mask_inv) cv2.error: /Users/travis/build/skvark/opencv-python/opencv/modules/core/src/arithm.cpp:241: error: (-215) (mtype == 0 || mtype == 1) && _mask.sameSize(*psrc1) in function binary_op
Better exception is at least required. Best solution would probably be automatic rescaling of the sub image (cigar/glasses).