How do I remove background from an image like this?-CodePudding

The I want to remove the background, and draw the outline of the box shown in the image(there are

Image 2:

Notes

Note that the processed image (which is binary) is inverted at cv2.bitwise_not(img_erode). Observe the processed version of both images (returned by the process() function defined above), with the inversion:

Processed Image 1:

Processed Image 2:

Tools

Finally, if you happen to have other images where the above program doesn't work properly on, you can use

CodePudding user response：

I would suggest you to label the dataset manually and then train an image segmentation model. To label the dataset, you can use tools like CVAT or labelme. For the model, consider using U-net.