poppinace / indexnet_matting

(ICCV'19) Indices Matter: Learning to Index for Deep Image Matting
Other
389 stars 90 forks source link

training from scratch #13

Closed liminn closed 4 years ago

liminn commented 4 years ago

Thanks for great work~ I am trying to training your model from scratch(only use pretrained mobilenet weights). I encountered two problems:

屏幕快照 2019-11-27 下午8 57 15

For the first problem, I think my training epochs is not enough (just trained to the 6th epoch). For the second problem, I am very confused and have no ideas. Do you have any suggestions on these two problems?

poppinace commented 4 years ago

Hi @liminn,

In my opinion, I think your ground truth alpha mattes are not generately correctly. Please check your code making sure the alpha is within the range [0, 1].

Hao

liminn commented 4 years ago

@poppinace Thanks, I will check the alpha labels.

facetohard commented 4 years ago

Hello @liminn @poppinace , I am trying to train indexnet from scratch. But I got worse performance than deep image matting by using the paper learning rate config。 So I wonder that how you trained indexnet. Does you train with multi stage just like the paper "Deep Image Matting" does.