azaion/ai-training
1
0
Fork 0
mirror of https://github.com/azaion/ai-training.git synced 2026-04-22 09:16:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

GPU augmentation. try2

Browse Source
This commit is contained in:
zxsanny
2025-03-05 10:48:45 +02:00
parent 5bd519b968
commit 2611a22264
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
preprocessing-cuda.py
+2 -2
View File
@@ -62,8 +62,8 @@ class DataLoader:
aug_image = fn.random_resized_crop( aug_image = fn.random_resized_crop(
images, images,
device='gpu', device='gpu',
min_scale=0.8, size=(images.shape[1], images.shape[2]),
max_scale=1.0 random_area=(0.8, 1.0)
) )
# Apply multiple random augmentations # Apply multiple random augmentations