This repository is no longer maintained, all migrated to BSDA.
Bayesian Random Semantic Data Augmentation (BRSDA) algorithm implement in Pytorch。
from .brsda_warp import BRSDAWarp
model = Backbone()
brsda_warp = BRSDAWarp(model, num_classes, multi=10, lambda=0.8)
...
# Training
outputs = brsda_warp(x, is_train=True)
loss = brsda_warp.get_loss(outputs, targets, criterion, is_train=True)
# forward loss
...