diff --git a/htc_ls_model_all.py b/htc_ls_model_all.py index 94aa0dd2d80c67acba1e427b616a48ad50bd2895..f7f5398e5f2a97a35796ce87e4c1249f5055aea0 100644 --- a/htc_ls_model_all.py +++ b/htc_ls_model_all.py @@ -697,7 +697,7 @@ data = dict( dict(type='LoadImageFromFile'), dict( type='MultiScaleFlipAug', - img_scale=[(850, 850)], + img_scale=[(1080, 720)], flip=False, transforms=[ dict(type='Resize', keep_ratio=True), diff --git a/new_eval_all.py b/new_eval_all.py index 446a46e85e38fcc376a80f2b878876bae39b820f..eb69611d1a702b7f2dbd2ae93f2622dbb9a47ca9 100644 --- a/new_eval_all.py +++ b/new_eval_all.py @@ -242,7 +242,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")