-
Notifications
You must be signed in to change notification settings - Fork 379
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add xView2 Dataset #236
Add xView2 Dataset #236
Conversation
) | ||
|
||
|
||
ColorMap = Union[List[Union[str, Tuple[int, int, int]]], str, Tuple[int, int, int]] |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@adamjstewart sphinx docs builds are failing due to the below errors. Any idea?
|
@isaaccorley this is from the intersphinx mapping, fix is here Line 102 in 41502e1
|
6702954
to
a96f4a8
Compare
@adamjstewart @isaaccorley I think this is finished now. Some notes:
|
|
||
|
||
class TestXView2: | ||
@pytest.fixture(params=["train", "test"]) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@pytest.fixture(params=["train", "test"]) | |
@pytest.fixture(scope="class", params=["train", "test"]) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This breaks the tests
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
ScopeMismatch: You tried to access the 'function' scoped fixture 'monkeypatch' with a 'class' scoped request object, involved factories
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
This PR adds the xView2 train/test set used in the xView2 challenge and described in "xBD: A Dataset for Assessing Building Damage from Satellite Imagery", Gupta et al. (2019)
This allows for a user to load the train/test 1024x1024 imagery and masks as a Vision Dataset, however in the homepage there are additional download links for non chipped imagery and the geojson labels which we should consider adding later as a Geo Dataset. This dataset cannot be automatically downloaded as you must register at the xView2 homepage and the download links expire after a few minutes.
draw_semantic_segmentation_mask
method totorchgeo.datasets.utils
which takes an image tensor, semantic mask tensor (h, w), alpha blend factor, and colormap and overlays the mask onto the image, returning a numpy array which can be used for plottingTODO:
- plot examplesExamples: