| 12345678910111213141516171819202122232425 |
- # Copyright (c) OpenMMLab. All rights reserved.
- from mmdet.registry import DATASETS
- from .coco import CocoDataset
- @DATASETS.register_module()
- class iSAIDDataset(CocoDataset):
- """Dataset for iSAID instance segmentation.
- iSAID: A Large-scale Dataset for Instance Segmentation
- in Aerial Images.
- For more detail, please refer to "projects/iSAID/README.md"
- """
- METAINFO = dict(
- classes=('background', 'ship', 'store_tank', 'baseball_diamond',
- 'tennis_court', 'basketball_court', 'Ground_Track_Field',
- 'Bridge', 'Large_Vehicle', 'Small_Vehicle', 'Helicopter',
- 'Swimming_pool', 'Roundabout', 'Soccer_ball_field', 'plane',
- 'Harbor'),
- palette=[(0, 0, 0), (0, 0, 63), (0, 63, 63), (0, 63, 0), (0, 63, 127),
- (0, 63, 191), (0, 63, 255), (0, 127, 63), (0, 127, 127),
- (0, 0, 127), (0, 0, 191), (0, 0, 255), (0, 191, 127),
- (0, 127, 191), (0, 127, 255), (0, 100, 155)])
|
