curioustorvald/Terrarum-sans-bitmap
1
0
Fork 0
mirror of https://github.com/curioustorvald/Terrarum-sans-bitmap.git synced 2026-06-06 05:58:30 +09:00
Code Issues Packages Projects Releases Wiki Activity

revised autokem model

Browse Source
This commit is contained in:
minjaesong
2026-03-09 23:54:51 +09:00
parent 268610a8b3
commit 71370a82a7
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Autokem/train_torch.py
View File

@@ -398,7 +398,7 @@ def _shift_image(img, dx, dy):
def _augment_one(img, label, rng): def _augment_one(img, label, rng):
"""One augmented copy: random 1px shift + 1% pixel dropout.""" """One augmented copy: random 1px shift + 1% pixel dropout."""
dx = rng.integers(-1, 2) # -1, 0, or 1 dx = rng.integers(-1, 2) # -1, 0, or 1
dy = 0 dy = rng.integers(-1, 2) # -1, 0, or 1
aug = _shift_image(img, dx, dy) aug = _shift_image(img, dx, dy)
# mask = rng.random(aug.shape) > 0.01 # mask = rng.random(aug.shape) > 0.01
# aug = aug * mask # aug = aug * mask