|
|
|
|
|
|
|
|
|
import numpy as np |
|
|
|
__all__ = ["rand_bbox"] |
|
|
|
|
|
def rand_bbox( |
|
h: int, |
|
w: int, |
|
lam: float, |
|
rand_func: callable = np.random.uniform, |
|
) -> tuple[int, int, int, int]: |
|
"""randomly sample bbox, used in cutmix""" |
|
cut_rat = np.sqrt(1.0 - lam) |
|
cut_w = w * cut_rat |
|
cut_h = h * cut_rat |
|
|
|
|
|
cx = rand_func(0, w) |
|
cy = rand_func(0, h) |
|
|
|
bbx1 = int(np.clip(cx - cut_w / 2, 0, w)) |
|
bby1 = int(np.clip(cy - cut_h / 2, 0, h)) |
|
bbx2 = int(np.clip(cx + cut_w / 2, 0, w)) |
|
bby2 = int(np.clip(cy + cut_h / 2, 0, h)) |
|
|
|
return bbx1, bby1, bbx2, bby2 |
|
|
