Граф коммитов

581 Коммитов

Автор SHA1 Сообщение Дата
Isaac Corley 1a49cfffbe update tests 2021-09-28 12:43:27 -07:00
Isaac Corley 2ff7d776e7 refactor to inherit VisionClassificationDataset 2021-09-28 12:43:27 -07:00
Ashwin Nair 1c833d5d7e
SpaceNet 2 (#155)
* Add Spacenet2

* Fix logical errors

* Use label_glob

* Refactor Spacenet

* Add tests and test data for SpaceNet2

* Make SpaceNet an ABC

* Bugfix

* Add docstring for SpaceNet metaclass

* Formatting

* Add metaclass to docs

* Hack for sn2_AOI_2_Vegas

* Add missing tests

* Change md5 hash for test data

* Update spacenet.py

* Fix regex pattern

* Update regex pattern

* Fixes

* Indent-fix

* Remove nested list

* Cleanup

Co-authored-by: Caleb Robinson <calebrob6@gmail.com>
2021-09-28 10:55:56 -05:00
isaac a3b636fe99
Add VisionClassificationDataset (#171)
* updated docs

* added VisionClassificationDataset

* refactor PatternNet and RESISC45 to inherit VisionClassificationDataset

* added unit tests and sample data

* refactor PatternNet and RESISC45 to new download/verify checks and fix code coverage

* remove override of __str__

* set default_loader as loader default

* removed loader arg from datasets

* update tests

* format

* remove duplicate code

* updated docstrings
2021-09-27 20:55:50 -05:00
Caleb Robinson 79476bb42a
Added the UC Merced dataset (#169)
* Added UC Merced dataset

* Added UC Merced dataset

* Adding docs and classes to docstring

* Addressing review
2021-09-27 20:14:02 -05:00
Caleb Robinson 489ffdc2bd
Added EuroSat dataset (#167)
* Added EuroSat dataset

* Cleaning up

* Removing unzipped data

* Added to docs

* EuroSat --> EuroSAT to match paper

* Changing class listing to use bullets

* Update torchgeo/datasets/eurosat.py

Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>

* Addressing review

Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
2021-09-27 20:13:17 -05:00
Adam J. Stewart 5d4ad430d2
nbsphinx in-line code links are broken 2021-09-26 11:25:28 -05:00
Caleb Robinson 98f6af7757 Formatting 2021-09-26 01:55:56 -07:00
Caleb Robinson aff8bd31e2
Disabling requirejs in nbsphinx to fix docs 2021-09-26 01:49:28 -07:00
Caleb Robinson a86dbe6f7e
Tutorials: use pip install torchgeo (#164) 2021-09-25 15:25:30 -05:00
Ashwin Nair 68d9d4a43d
blk-fix (#165) 2021-09-25 00:43:06 -07:00
Caleb Robinson fcd7279dbd Making the Colab badge actually work 2021-09-24 23:39:42 -07:00
Caleb Robinson f14d5986ca The big colab badge had to go 2021-09-24 23:34:25 -07:00
Adam J. Stewart 9911afef89 Minor grammar fixes 2021-09-24 16:56:47 -07:00
Caleb Robinson 10508dcad4 Renamed trainer tutorial 2021-09-24 16:56:47 -07:00
Caleb Robinson b3e468641f Renamed trainer tutorial 2021-09-24 16:56:47 -07:00
Caleb Robinson 59c3593bad Updated trainer notebook 2021-09-24 16:56:47 -07:00
Caleb Robinson d04fc4bd43 Added tutorial to the docs 2021-09-24 16:56:47 -07:00
Caleb Robinson c800dd1de3 Adding Cyclone trainer tutorial 2021-09-24 16:56:47 -07:00
Zhuo Zheng 78fe113ce5
Adding ChangeStar model (#157)
* add FarSeg model

* Revised.

* Ran black and working on mypy fixes

* Fixed other mypy issues

* try to fix mypy issues

* Fixing rest of mypy issues

* try to fix mypy issues

* Fixed issue with in_channels and backbone_pretrained

* Removing change to in_channels, formatting doc

* revised.

- improving the docstring of FarSeg
- removing the scale_aware_proj option in FSRelation class
- removing the in_channels option in FarSeg class
- improving the test_farseg.py by parametrization

* changestar

* removing the function `backbone_forward`

* update ChangeStar

* update changestar

* fixed a typo in docstring

* fixing mypy, pytest, sphinx issues

* revised.

major:
- removing unused import
- split the testing case
- removing the redundant docstring

* fixing mypy and sphinx issues

major:
- adding more description for docstring of ChangeStar

* fixing mypy and sphinx issues

* fixing mypy and sphinx issues

* fixing mypy issues

* Update torchgeo/models/changestar.py

Co-authored-by: isaac <22203655+isaaccorley@users.noreply.github.com>

* Update torchgeo/models/changestar.py

Co-authored-by: isaac <22203655+isaaccorley@users.noreply.github.com>

* revised.

* fixing flake8 issues

* fixing mypy issues.

- replacing Tensor with Any, because a list of tensor as value stored in dictionary.

* update ChangeStar

* fixing mypy issues

* fixing mypy issues

* fixing mypy issues

* Adding more docstring about how to use ChangeStar for multi-class change detection

* Adding missing test cases

* revised.

- replacing Dict[str, Any] with Dict[str, Tensor]

Co-authored-by: Caleb Robinson <calebrob6@gmail.com>
Co-authored-by: isaac <22203655+isaaccorley@users.noreply.github.com>
2021-09-24 03:36:11 +00:00
Adam J. Stewart 6d5725056c
Fix broken section links in Contributing docs 2021-09-22 13:45:44 -05:00
Adam J. Stewart 8c450b0b6b Use docs/requirements.txt to list deps 2021-09-22 09:20:58 -07:00
Adam J. Stewart cc53ee2e8a Move contribution guide to documentation 2021-09-22 09:20:58 -07:00
Caleb Robinson 682c35b94a Updates to pre-commit section 2021-09-22 09:20:58 -07:00
Caleb Robinson 2c48eea46b Actually added pre-commit instructions 2021-09-22 09:20:58 -07:00
Caleb Robinson b49bda03a1 Adding instructions for new contributors 2021-09-22 09:20:58 -07:00
Adam J. Stewart 04c754fddb
Add dataset/sampler benchmarking script (#115)
* Add dataset/sampler benchmarking script

* Some changes to get the benchmark script working

* Added writing results to file

* Added script for running a grid of benchmark experiments

* Actual experiment configuration

* Improve help message formatting

* Remove default for mutually exclusive required group

* Rounding, units, ignore output file

* Fix a couple bugs in counting patches

* Display cache info

* Increase cache size

* Cache is shared

* Benchmark model as well

* Warp to same CRS/res as CDL

* Work around bug in sampler

* Fix isort

* Allow specification of CPU/GPU device

Co-authored-by: Caleb Robinson <calebrob6@gmail.com>
2021-09-22 09:47:25 -05:00
Adam J. Stewart 161e9de2cc
Add installation instructions (#153)
* Add installation instructions

* Comment out instructions that require a stable release
2021-09-22 09:41:52 -05:00
Adam J. Stewart 0820b2b131
Reduce number of files listing deps (#154)
* Reduce number of files listing deps

* Install cython before anything else

* Fix missing deps before install

* Fix typo in setup.cfg extra option

* Add Read the Docs config file

* nbmake only used for testing docs, not building docs
2021-09-22 09:41:00 -05:00
Zhuo Zheng 53ff5ffbe5
Add FarSeg model (#150)
* add FarSeg model

* Revised.

* Ran black and working on mypy fixes

* Fixed other mypy issues

* try to fix mypy issues

* Fixing rest of mypy issues

* try to fix mypy issues

* Fixed issue with in_channels and backbone_pretrained

* Removing change to in_channels, formatting doc

* revised.

- improving the docstring of FarSeg
- removing the scale_aware_proj option in FSRelation class
- removing the in_channels option in FarSeg class
- improving the test_farseg.py by parametrization

* removing the function `backbone_forward`

* Various improvements

Co-authored-by: Caleb Robinson <calebrob6@gmail.com>
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
2021-09-21 17:21:13 -05:00
Ashwin Nair 1298706aa7
Set src properties (#156) 2021-09-21 13:54:40 -05:00
Adam J. Stewart 907593abbc
Fix pip installation 2021-09-20 17:28:01 -05:00
Adam J. Stewart 43ab649c8f
Use __author__ field 2021-09-20 17:25:35 -05:00
Ashwin Nair 30a082d33c
Add download_radiant_mlhub_collection (#152)
* Add download_radiant_mlhub_collection

* Add tests

* Rename download_radiant_mlhub -> download_radiant_mlhub_dataset

* Update test_utils.py

* Update test_utils.py

* Update test_utils.py

Co-authored-by: Caleb Robinson <calebrob6@gmail.com>
2021-09-20 07:23:42 +00:00
isaac 459524fedc
Add ZueriCrop dataset (#147) 2021-09-19 23:25:09 +00:00
isaac 77094c21fa
Add ADVANCE dataset (#133)
* updated docs

* added torchaudio as optional dependency

* added sample data for tests

* added dataset

* added advance unit tests

* replaced torchaudio with scipy.io.wavefile.read

* Revert "added torchaudio as optional dependency"

This reverts commit 960d94f67533e83facc080d9a6f9b965a8820294.

* updated to lazy import scipy, updated docstring

* add pytest.importorskip check for scipy

* add sample audio wav file creation details

* add scipy dependency

* downgrading scipy dep to scipy>=1.5.4

* fix pytest.importorskip to return none

* update scipy import error message

* fixed dummy audio data dims

* downgrading scipy dep to scipy>=0.9.0

* added tests for missing h5py

* format

* fixed missing import test

* Update tests/datasets/test_advance.py

Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>

Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
2021-09-19 23:00:56 +00:00
Adam J. Stewart 58d05d5950
Remove Codecov annotations from PRs 2021-09-19 11:07:39 -05:00
Caleb Robinson 01b8c64e34
Feature/chesapeake cvpr training (#142)
* Nodata check

* Changing val to use grid sampler

* Adding focal loss option

* Added FCN to Chesapeake trainer

* Bug fix in loss arguments

* Added different HR class sets to chesapeake trainer and made the nodata values sensible

* Changing the chesapeake configs

* Changing the chesapeake configs

* Added TODO and simplified some logic
2021-09-17 16:51:18 -05:00
Adam J. Stewart b7c80793c9 Required checks must always be run 2021-09-17 14:39:19 -07:00
Adam J. Stewart 8a02183f62
CDL: don't re-download dataset if zip files are missing (#138)
* CDL: don't re-download dataset if zip files are missing

* Fix pydocstyle

* Reword checksum parameter
2021-09-16 16:22:10 +00:00
Adam J. Stewart 8c49b2a906
Proposal: better handling of partial timestamps (#134)
* Proposal: better handling of partial timestamps

* Parse format string directly

* Add unit tests

* Windows is broken

* Windows is still broken

* Fix mypy

* Simplify logic

* Fix bug for month 12, add details to docstring
2021-09-16 11:07:09 -05:00
Adam J. Stewart 73ba391bc8
Codecov upload token not needed for public repos 2021-09-15 20:41:43 -05:00
Ashwin Nair b48b5cd2ad
Merge pull request #137 from ashnair1/pre-commit
Add pre-commit config
2021-09-15 15:46:35 -07:00
Caleb Robinson a82d3f47b4
Cleaning up ChesapeakeCVPR (#131)
* Implemented way to test if ChesapeakeCVPR has been downloaded

* Cleaned up the way transforms are used in ChesapeakeCVPRDataModule and changed the test dataloader to use a GridGeoSampler

* Adding other transforms and cleaning up style

* Changing masks to mask throughout

* Organize the way metrics are logged

* Formatting

* Adding rasterio data loadering best practices and fixing trainer a bit

* Script for evaluating trained models over all states

* Forgot to update the MD5 after changing the dataset nodata masks

* Update test_chesapeakecvpr_models.py

Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>

* Changes recommended by Adam

* Idon'tsort

Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
2021-09-15 11:52:51 -05:00
Ashwin Nair 0ce0a591b6
Add Spacenet 1: Building Detection v1 (#129)
* Add Spacenet 1

* Add test data

* Style fixes

* Convert Spacenet1 to VisionDataset

* Add option for selecting imagery

* Consolidate spacenet

* Create single spacenet.py for all spacenet datasets
* Create single spacenet directory for all spacenet test data
* Create single test_spacenet.py for testing all spacenet datasets

* Add copyright

* Reorder Spacenet in docs

* Test both rgb & 8band

* Rename Spacenet -> SpaceNet
2021-09-15 11:35:15 -05:00
Adam J. Stewart 3313a36014
Fix typo in So2Sat docstring 2021-09-14 15:15:08 -05:00
Ashwin Nair ef0fe418e6
Add models to pydocstyle match_dir (#136) 2021-09-14 15:53:38 +00:00
Adam J. Stewart 84521dc977
Fix minimum setuptools versions 2021-09-14 10:42:34 -05:00
Caleb Robinson b6ccd18193
Adding a length property to GridGeoSampler (#128)
* Adding length to GridGeoSampler

* Factor out logic for sampling a random bounding box from a given bounds
2021-09-12 10:51:27 -05:00
isaac f60cbee39b
Add RESISC45 Dataset (#126)
* updated docs

* added sample data for tests

* added unit tests

* added dataset

* updated tests to not run on windows due to rar
2021-09-12 10:50:15 -05:00