数据增强

2026-02-19 21:29:32 +08:00
parent 88d14221ae
commit 7023d5dde4
3 changed files with 93 additions and 2 deletions
--- a/detr/main.py
+++ b/detr/main.py
@@ -73,6 +73,7 @@ def get_args_parser():
    parser.add_argument('--kl_weight', action='store', type=int, help='KL Weight', required=False)
    parser.add_argument('--chunk_size', action='store', type=int, help='chunk_size', required=False)
    parser.add_argument('--temporal_agg', action='store_true')
+    parser.add_argument('--image_aug', action='store_true')

    return parser