Skip to content

Commit

Permalink
Fix bugs
Browse files Browse the repository at this point in the history
  • Loading branch information
yuyi1005 committed Dec 6, 2023
1 parent 929a8ed commit 507a932
Show file tree
Hide file tree
Showing 2 changed files with 21 additions and 21 deletions.
4 changes: 2 additions & 2 deletions configs/_base_/datasets/dior.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
# dataset settings
dataset_type = 'DIORDataset'
data_root = 'data/dior/'
data_root = 'data/DIOR/'
backend_args = None

train_pipeline = [
Expand Down Expand Up @@ -60,7 +60,7 @@
backend_args=backend_args)
]))
val_dataloader = dict(
batch_size=8,
batch_size=1,
num_workers=2,
persistent_workers=True,
drop_last=False,
Expand Down
38 changes: 19 additions & 19 deletions configs/_base_/datasets/dota.py
Original file line number Diff line number Diff line change
Expand Up @@ -59,27 +59,27 @@
data_prefix=dict(img_path='trainval/images/'),
test_mode=True,
pipeline=val_pipeline))
# test_dataloader = val_dataloader
test_dataloader = val_dataloader

val_evaluator = dict(type='DOTAMetric', metric='mAP')
# test_evaluator = val_evaluator
test_evaluator = val_evaluator

# inference on test dataset and format the output results
# for submission. Note: the test set has no annotation.
test_dataloader = dict(
batch_size=8,
num_workers=2,
persistent_workers=True,
drop_last=False,
sampler=dict(type='DefaultSampler', shuffle=False),
dataset=dict(
type=dataset_type,
data_root=data_root,
data_prefix=dict(img_path='test/images/'),
test_mode=True,
pipeline=test_pipeline))
test_evaluator = dict(
type='DOTAMetric',
format_only=True,
merge_patches=True,
outfile_prefix='./work_dirs/dota/Task1')
# test_dataloader = dict(
# batch_size=1,
# num_workers=2,
# persistent_workers=True,
# drop_last=False,
# sampler=dict(type='DefaultSampler', shuffle=False),
# dataset=dict(
# type=dataset_type,
# data_root=data_root,
# data_prefix=dict(img_path='test/images/'),
# test_mode=True,
# pipeline=test_pipeline))
# test_evaluator = dict(
# type='DOTAMetric',
# format_only=True,
# merge_patches=True,
# outfile_prefix='./work_dirs/dota/Task1')

0 comments on commit 507a932

Please sign in to comment.