Commit 25bae578 authored by nikhil_rayaprolu's avatar nikhil_rayaprolu

Update htc_r50.py

parent 1d97912f
...@@ -181,7 +181,7 @@ test_cfg = dict( ...@@ -181,7 +181,7 @@ test_cfg = dict(
keep_all_stages=False) keep_all_stages=False)
# dataset settings # dataset settings
dataset_type = 'CocoDataset' dataset_type = 'CocoDataset'
data_root = '/ssd_scratch/cvit/nikhilnew/crowd/data/' data_root = 'data/coco/'
img_norm_cfg = dict( img_norm_cfg = dict(
mean=[123.675, 116.28, 103.53], std=[58.395, 57.12, 57.375], to_rgb=True) mean=[123.675, 116.28, 103.53], std=[58.395, 57.12, 57.375], to_rgb=True)
albu_train_transforms = [ albu_train_transforms = [
...@@ -312,6 +312,6 @@ total_epochs = 30 ...@@ -312,6 +312,6 @@ total_epochs = 30
dist_params = dict(backend='nccl') dist_params = dict(backend='nccl')
log_level = 'INFO' log_level = 'INFO'
work_dir = './work_dirs/htc_r50_fpn' work_dir = './work_dirs/htc_r50_fpn'
load_from = '/home/vamsi.muthireddy/htc_r50_fpn_20e_20190408-c03b7015.pth' load_from = 'htc_r50_fpn_20e_20190408-c03b7015.pth'
resume_from = None resume_from = None
workflow = [('train', 1)] workflow = [('train', 1)]
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment