NVlabs/DeepInversion

loss about segmentation task

raninbowlalala opened this issue · 3 comments

Thanks for your work!
I want to generate images from segmentaion task model(like deeplabv3plus), but I use your original loss got bad result.
Could you help me with this issue?
Thanks a lot!

Other networks like object detector requires augmentations: please check the new WACV'21 paper here. There may be room for segmentation networks too.

Other networks like object detector requires augmentations: please check the new WACV'21 paper here. There may be room for segmentation networks too.

Thank you very much!
I want to try the data augmentation you mentioned- cutout. Could you share with me about what minsize, maxsize and cut number you use?
Do you have plan to opensource the code?

Hi @raninbowlalala we have released the code for DI for object detection here: https://github.com/NVlabs/DIODE let me know if it helps.