Good pretrained pytorch background removal model

[kq3rHCt9gK]

Removing background of image (pytorch, python)

Link to the jupyter notebook containing my code is here: https://github.com/alpizano/machine-learning-neural-network/blob/master/image_processing.ipynb So I am starting a DL neural network project and my Professor assigned that we FIRST take an image in python (+pytorch) and convert 3d matrix (rows, cols, RGB) to 2d matrix (rows,cols) by averaging (R+G+B)/3 channels to turn into grayscale, THEN to remove the background of the image by picking an intensity of a pixel, say 200, and looping/iterat…

It seems that robust background removal was still challenging.

But it seems the service offered by this website is OK. Is there a pretrained background removal model that has a similar performance as this website.

Remove Background from Image for Free – remove.bg