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))
|