提交 df31d808 编写于 作者: Y Yang Nie 提交者: Tingquan Gao

bugfix: MixupOperator, CutmixOperator, FmixOperator

上级 cabdc251
...@@ -88,7 +88,7 @@ class MixupOperator(BatchOperator): ...@@ -88,7 +88,7 @@ class MixupOperator(BatchOperator):
def __call__(self, batch): def __call__(self, batch):
imgs, labels, bs = self._unpack(batch) imgs, labels, bs = self._unpack(batch)
idx = np.random.permutation(bs) idx = np.arange(bs)[::-1]
lam = np.random.beta(self._alpha, self._alpha) lam = np.random.beta(self._alpha, self._alpha)
imgs = lam * imgs + (1 - lam) * imgs[idx] imgs = lam * imgs + (1 - lam) * imgs[idx]
targets = self._mix_target(labels, labels[idx], lam) targets = self._mix_target(labels, labels[idx], lam)
...@@ -143,7 +143,7 @@ class CutmixOperator(BatchOperator): ...@@ -143,7 +143,7 @@ class CutmixOperator(BatchOperator):
def __call__(self, batch): def __call__(self, batch):
imgs, labels, bs = self._unpack(batch) imgs, labels, bs = self._unpack(batch)
idx = np.random.permutation(bs) idx = np.arange(bs)[::-1]
lam = np.random.beta(self._alpha, self._alpha) lam = np.random.beta(self._alpha, self._alpha)
bbx1, bby1, bbx2, bby2 = self._rand_bbox(imgs.shape, lam) bbx1, bby1, bbx2, bby2 = self._rand_bbox(imgs.shape, lam)
...@@ -179,7 +179,7 @@ class FmixOperator(BatchOperator): ...@@ -179,7 +179,7 @@ class FmixOperator(BatchOperator):
def __call__(self, batch): def __call__(self, batch):
imgs, labels, bs = self._unpack(batch) imgs, labels, bs = self._unpack(batch)
idx = np.random.permutation(bs) idx = np.arange(bs)[::-1]
size = (imgs.shape[2], imgs.shape[3]) size = (imgs.shape[2], imgs.shape[3])
lam, mask = sample_mask(self._alpha, self._decay_power, \ lam, mask = sample_mask(self._alpha, self._decay_power, \
size, self._max_soft, self._reformulate) size, self._max_soft, self._reformulate)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册