From de6290a2dcf83f9f9981489ec6f98807c9e4d16b Mon Sep 17 00:00:00 2001
From: shraddhaamohan <shraddhaa.mohan@gmail.com>
Date: Fri, 8 Jan 2021 04:51:40 +0530
Subject: [PATCH] 1080,720 flip false 0.0015 thr nms 0.8
---
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 94aa0dd..f7f5398 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 446a46e..eb69611 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")
--
GitLab