Pytorch implementation of our paper.
Satoru Mizusawa and Yuichi Sei: Interlayer Augmentation in a Classification Task, 4th IEEE International Conference on Computing, Electronics & Communications Engineering (iCCECE), pp.59-64 (2021.8)
pip install git+https://github.com/bottlenome/rbg.git
see train.py
- import RBG or BG
from rbg.generalization import RandomBatchGeneralization, BatchGeneralization - Apply to model
See rgb/model.py WrapVGG