shvardhan commited on
Commit
6c33fe7
·
1 Parent(s): a3a1e0f

update config file

Browse files
configs/_base_/faster-rcnn_r50_fpn_1x_coco.py CHANGED
@@ -198,21 +198,31 @@ test_pipeline = [
198
  dict(type='Collect', keys=['img']),
199
  ])
200
  ]
 
 
 
 
 
 
 
201
  data = dict(
202
  samples_per_gpu=2,
203
  workers_per_gpu=2,
204
  train=dict(
205
  type=dataset_type,
 
206
  ann_file=data_root + 'train.json',
207
  img_prefix=data_root + 'train/',
208
  pipeline=train_pipeline),
209
  val=dict(
210
  type=dataset_type,
 
211
  ann_file=data_root + 'val.json',
212
  img_prefix=data_root + 'val/',
213
  pipeline=test_pipeline),
214
  test=dict(
215
  type=dataset_type,
 
216
  ann_file=data_root + 'val.json',
217
  img_prefix=data_root + 'val/',
218
  pipeline=test_pipeline))
 
198
  dict(type='Collect', keys=['img']),
199
  ])
200
  ]
201
+
202
+ metainfo = {
203
+ 'classes': ('orgaquant', ),
204
+ 'palette': [
205
+ (220, 20, 60),
206
+ ]
207
+ }
208
  data = dict(
209
  samples_per_gpu=2,
210
  workers_per_gpu=2,
211
  train=dict(
212
  type=dataset_type,
213
+ metainfo = metainfo,
214
  ann_file=data_root + 'train.json',
215
  img_prefix=data_root + 'train/',
216
  pipeline=train_pipeline),
217
  val=dict(
218
  type=dataset_type,
219
+ metainfo = metainfo,
220
  ann_file=data_root + 'val.json',
221
  img_prefix=data_root + 'val/',
222
  pipeline=test_pipeline),
223
  test=dict(
224
  type=dataset_type,
225
+ metainfo = metainfo,
226
  ann_file=data_root + 'val.json',
227
  img_prefix=data_root + 'val/',
228
  pipeline=test_pipeline))