search

SkyTNT / anime-segmentation

high-accuracy segmentation for anime character
Apache License 2.0
649 stars 62 forks source link

InSPyReNet #7

Closed Zarxrax closed 1 year ago

Zarxrax commented 1 year ago

Is it possible to add support for InSPyReNet to the training code? https://github.com/plemeri/inspyrenet

Their paper shows significant improvement from ISNet.

SkyTNT commented 1 year ago

Added it just now. But i have no time to train it.

Zarxrax commented 1 year ago

Thanks, I was able to train it for a few hours to test. In train.py, I had to revert line 188 back to: strategy="ddp_find_unused_parameters_false" if opt.devices > 1 else None, Then the training worked.

I was not able to get export to onnx working. But I could test through the app.py. So far it does not seem to be as good as the other models, but I will try some experiments to get a better result.