From 21acc1059a6301bfbd1ff965976a060b893d8f02 Mon Sep 17 00:00:00 2001
From: shraddhaamohan <shraddhaa.mohan@gmail.com>
Date: Sat, 9 Jan 2021 01:47:39 +0530
Subject: [PATCH] e24 1080,720 flip false 0.0015 scr thr 0.8 nms

---
 htc_ls_model_all.py | 6 +++---
 new_eval_all.py     | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/htc_ls_model_all.py b/htc_ls_model_all.py
index 12d9225..f7f5398 100644
--- a/htc_ls_model_all.py
+++ b/htc_ls_model_all.py
@@ -217,9 +217,9 @@ test_cfg = dict(
         nms_thr=0.8,
         min_bbox_size=0),
     rcnn=dict(
-        score_thr=0.002,
+        score_thr=0.0015,
         nms=dict(type='nms', iou_thr=0.35),
-        max_per_img=50,
+        max_per_img=100,
         mask_thr_binary=0.4))
 classes = [
     'water', 'pear', 'egg', 'grapes', 'butter', 'bread-white', 'jam',
@@ -697,7 +697,7 @@ data = dict(
             dict(type='LoadImageFromFile'),
             dict(
                 type='MultiScaleFlipAug',
-                img_scale=[(1200, 1960)],
+                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 d4c4cc1..7b2ce54 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_23_all.pth", map_location='cpu')
+        checkpoint = load_checkpoint(model,"models/epoch_24_all.pth", map_location='cpu')
 
 
         json_file = open("data/edited_val_r3.json")
-- 
GitLab