Add Back Augmentations - Githubissues

leonardoaraujosantos / ChangeNet

Implementation of the ChangeNet paper

76 stars 20 forks source link

Add Back Augmentations #2

Open leonardoaraujosantos opened 5 years ago

leonardoaraujosantos commented 5 years ago

References

Pytorch Issue Message
Pytorch Discussion
Pytorch Discussion 2

leonardoaraujosantos commented 4 years ago

I've added back some augmentations, Including the swap (reference, test) which is confirmed that does NOT help on the current architecture, also unfreezing the Resnet branches gave 12%+ accuracy boost on the validation set.

Labels

Orange: Baseline with Resnets frozen
Magenta: Augmentations with Resnet unfrozen
Green: Swap with Augmentations and Resnet unfrozen

TODO: Still add the flip on: (Reference, Test, and label) with same probability