Spaces:
Runtime error
Runtime error
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))
|