shape miss，After Update（4.8）

[jjandnn]

CUDA Device: 0
Using image mode
Traceback (most recent call last):
File "test_background-matting_image.py", line 179, in
comp_im_tr1=composite4(fg_out0,back_img10,alpha_out0)
File "/content/drive/My Drive/Background-Matting/functions.py", line 8, in composite4
im = alpha * fg + (1 - alpha) * bg
ValueError: operands could not be broadcast together with shapes (1080,1920,1) (1080,1980,3)

background/0002.png miss.
0001.png is ok.

[senguptaumd]

Error with only 0002.png, you mean? It works with 0001.png, right?

I will fix it soon. Thanks for bringing it to notice.

[jjandnn]

Error with only 0002.png。
The actual size of 0002 is 1920 * 1047. I changed it to 1920 * 1080 and it is not wrong. I found that the pictures related to _back, including the background picture, as long as the size is not 1920 * 1080, it will not work, it will report an error.
thanks。
Your project is really great. Good luck。

[senguptaumd]

Thanks for the detailed bug report. Fixed it :)

[jjandnn]

The boss hard！