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