How does DeepMosaics recognize mosaics?

[taeyeong132]

First of all, thank you very much for making a great project.

Question

---

DeepMosaics seems to recognize the mosaic area and process only that part,
so I wonder the way of recognize the mosaic area.

Please let me know if there are any codes or related documents that I need to see.

Thank you.

[wywywywy]

It uses a BiSeNet to get the mask. The checkpoint is the ./pretrained_models/mosaic/mosaic_position.pth

[ferograd]

Is there a way to refine it? And train it more?

Some of the mosaics are rather small, but would be really useful to get rid of regardless