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

816 Коммитов

Автор SHA1 Сообщение Дата
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
Caleb Robinson 455ea7e24b
Updated MD5 for the newest version of the Chesapeake dataset 2021-09-11 19:41:38 -07:00
isaac 67f7d8a520
Add ETCI2021 Dataset (#119)
* add dataset to docs

* add sample test data

* add dataset unit tests

* add etci2021 dataset

* updated tests

* updated dataset to download only desired split file

* removed flood mask from file list for test set and other formatting

* Update torchgeo/datasets/etci2021.py

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

* fixed doc formatting

Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
2021-09-11 20:05:38 -05:00
Isaac Corley ea7dc26393 updated link from arxiv to doi 2021-09-10 23:18:15 -05:00
Isaac Corley 2bc997dda0 format docstrings 2021-09-10 23:18:15 -05:00
Isaac Corley 19c06cebfe format and removed empty masks from file list 2021-09-10 14:49:19 -05:00
Isaac Corley c93b4ba21b added dataset 2021-09-10 14:49:19 -05:00
Isaac Corley 7de01364c9 added tests 2021-09-10 14:49:19 -05:00
Isaac Corley ba4cd3f75d added dummy sample test data 2021-09-10 14:49:19 -05:00
Isaac Corley 93d17804eb updated docs 2021-09-10 14:49:19 -05:00
Caleb Robinson cb7f0c9039
Fixed typo 2021-09-10 10:21:16 -07:00
Caleb Robinson cb3d7e13aa Some docstrings 2021-09-09 15:12:22 -07:00
Caleb Robinson aac2199cd9 Getting rid of pin_memory=False and proper typing of kwargs 2021-09-09 15:12:22 -07:00
Caleb Robinson c42171f75e Adding an empty line to conf files 2021-09-09 15:12:22 -07:00
Caleb Robinson 7f9dcf3701 Changes to fix Cyclone 2021-09-09 15:12:22 -07:00
Caleb Robinson 93b0dacc4a Some modifications to conf files 2021-09-09 15:12:22 -07:00