UNet для собственного набора данных RGB для удаления фона в кератах

Question

У меня 1000 изображений кошек с фоном. Я хочу попробовать UNet, чтобы удалить фон из этих изображений. Любые рекомендации приветствуются.

Должен ли я создавать аннотации или другие файлы, такие как файлы Json? Я не нашел подходящего способа сделать это.

Answer 1

Один из отличных способов реализовать удаление фона - это методы end2end, использующие преобразование изображений в изображения cGAN или pix2pix (вы можете использовать U-Net в вашей архитектуре gan)

Я настоятельно рекомендую вам прочитать эту статью: https://github.com/eti-p-doray/unet-gan-matting/blob/master/exploringImageMattingReport.pdf

, а также вы можете использовать метод pix2pix:

https://phillipi.github.io/pix2pix

http://www.k4ai.com/imageops/index.html