From a7574a3b5f710e2450853ba27cdd69b9ad3f313b Mon Sep 17 00:00:00 2001
From: shraddhaamohan <shraddhaa.mohan@gmail.com>
Date: Sat, 16 Jan 2021 15:35:59 +0530
Subject: [PATCH] round 4 e23 flip true 1080,720
---
htc_ls_model_all.py | 2 +-
new_eval_all.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/htc_ls_model_all.py b/htc_ls_model_all.py
index f7f5398..2414ff6 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 7b2ce54..d4c4cc1 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")
--
GitLab