train_tensorflow_orientation.py: size for crop

odulcy-mindee · Apr 16, 2024 · 1049fab · 1049fab
1 parent ea58710
commit 1049fab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/references/classification/train_tensorflow_orientation.py b/references/classification/train_tensorflow_orientation.py
@@ -147,7 +147,7 @@ def main(args):
     if not isinstance(args.workers, int):
         args.workers = min(16, mp.cpu_count())
 
-    input_size = (256, 256) if args.type == "page" else (32, 32)
+    input_size = (512, 512) if args.type == "page" else (128, 128)
 
     # AMP
     if args.amp: