diff --git a/htc_ls_model_all.py b/htc_ls_model_all.py index f7f5398e5f2a97a35796ce87e4c1249f5055aea0..2414ff6c9e6807ad5d0417d33a744493985adfca 100644 --- a/htc_ls_model_all.py +++ b/htc_ls_model_all.py @@ -698,7 +698,7 @@ data = dict( dict( type='MultiScaleFlipAug', img_scale=[(1080, 720)], - flip=False, + flip=True, transforms=[ dict(type='Resize', keep_ratio=True), dict(type='RandomFlip'), diff --git a/new_eval_all.py b/new_eval_all.py index 7b2ce549ded02868321fd6616e6107044e9704ad..d4c4cc1277e01ba738abb895437d10261d6f25bf 100644 --- a/new_eval_all.py +++ b/new_eval_all.py @@ -243,7 +243,7 @@ def evaluate(IMAGE_DIR, PREDICTION_FILE): if fp16_cfg is not None: wrap_fp16_model(model) - checkpoint = load_checkpoint(model,"models/epoch_24_all.pth", map_location='cpu') + checkpoint = load_checkpoint(model,"models/epoch_23_all.pth", map_location='cpu') json_file = open("data/edited_val_r3.json")