microsoft
/
torchgeo
зеркало из https://github.com/microsoft/torchgeo.git
1
0
Форкнуть 0
Код Задачи Пакеты Проекты Релизы Вики Активность
679 коммитов 12 Ветки 13 Теги 787 MiB
Граф коммитов

679 Коммитов

Автор SHA1 Сообщение Дата
Adam J. Stewart 13847d67fa
Remove sphinx CI test (#292) 
* Remove sphinx CI test

* Move mypy back to tests section

* Fix bad rebase
 2021-12-19 11:14:17 -06:00
Adam J. Stewart 01ae2db287
Rework list of required dependencies (#287) 
* Rework list of required dependencies

* Update open3d import error msg

* Style fixes

* Remove extra empty line

* Increase test coverage

* Fix idtrees tests
 2021-12-19 11:12:48 -06:00
Adam J. Stewart ae11f10502
Chesapeake: support already downloaded zip files (#281) 
* Chesapeake: support already downloaded zip files

* Remove comments, save for a later PR

* Style fix
 2021-12-19 11:07:44 -06:00
Ashwin Nair 4f6da2de6e
Replace pytorch-gpu with pytorch in environment.yml (#295) 
* Replace pytorch-gpu with pytorch in environment.yml

* Add windows note
 2021-12-19 11:05:35 -06:00
Aleksey Morozov 97877fbf99
Edit tutorials notebooks (#267) 
* remove unused import

* remove unused import

* remove unused imports in indices.ipynb

* remove unused imports and fix parameter in form cell in transforms.ipynb

* Revert "remove unused import"

This reverts commit 722fa90d9c969145f8abd402cf628dab7300914d.
 2021-12-19 10:56:50 -06:00
isaac e1c39e27d2
update for kornia v0.6.2 release bbox fix in K.AugmentationSequential (#268) 2021-12-19 10:56:20 -06:00
Adam J. Stewart 9a4868e4b1
Add ResNet pre-trained models to the docs (#256) 
* Add ResNet pre-trained models to the docs

* Fix missing reference link

* ResNet is not documented, nothing to link to
 2021-12-19 10:55:58 -06:00
Caleb Robinson 8501eea7e6
GitHub Actions: use Agg backend for matplotlib (#261) 2021-12-19 10:55:24 -06:00
Ashwin Nair 1679d9ed5e
Fix #230 (#249) 
* Move sklearn to dataset optional reqs

* Move sklean to mandatory reqs
 2021-12-19 10:55:05 -06:00
Adam J. Stewart 98140b56ef
VisionClassificationDataset: fix Sphinx base class mapping (#239) 
* VisionClassificationDataset: fix Sphinx base class mapping

* Import ImageFolder directly from where it is documented

* Override ImageFolder.__module__
 2021-12-19 10:54:38 -06:00
Caleb Robinson 430ea0d413
ETIC dataset was returning random samples for a given index 2021-12-19 10:52:16 -06:00
Adam J. Stewart 9f4120c6f4
Fix circular import issue 2021-12-19 10:50:50 -06:00
Adam J. Stewart 1dac56e8e0
Bump release version to 0.1.1 2021-12-19 10:45:50 -06:00
Adam J. Stewart 162b6bf493
Fix logo/author, include LICENSE in upload 2021-11-07 21:02:43 -06:00
Caleb Robinson 881dc9b349 Revert 2021-11-08 02:29:47 +00:00
Caleb Robinson 8a990c71df UCMerced should download by default in the trainers 2021-11-08 02:23:11 +00:00
Caleb Robinson 2b4f60dace Make the trainer notebook look stupid 2021-11-08 02:22:54 +00:00
Adam J. Stewart ea219cc0fe
Simplify tests a bit 2021-11-07 20:14:20 -06:00
Caleb Robinson 8d7f02aff6 yolo fix 2 pt. 1 2021-11-08 02:11:45 +00:00
Caleb Robinson cb90dfc53e yolo fix pt 2 2021-11-08 01:07:21 +00:00
Caleb Robinson b222b5b95b yolo fix 2021-11-08 01:06:00 +00:00
Caleb Robinson 43924f2eef Added correct weight string to UCMerced 2021-11-08 00:17:22 +00:00
Adam J. Stewart 3c26048582
Get back to 100% test coverage 2021-11-07 22:39:31 +00:00
Adam J. Stewart 5fae8c0982
Config file not being picked up 2021-11-07 22:23:32 +00:00
Adam J. Stewart 8ce9b6066e
Fix some bugs 2021-11-07 16:16:25 -06:00
Adam J. Stewart 49705cc215
Update pip installs in tutorials 2021-11-07 14:56:04 -06:00
Adam J. Stewart 8cc4d5e2c2
Various fixes to release tests 2021-11-07 14:52:52 -06:00
Adam J. Stewart 046f69a51b
Try again 2021-11-07 14:05:59 -06:00
Adam J. Stewart ad0439dbf5
Try different GDAL name 2021-11-07 14:03:22 -06:00
Adam J. Stewart 0f0687d411
Correct location for root_dir 2021-11-07 13:59:41 -06:00
Adam J. Stewart e9b8deed8a
Trainers should default to num_workers=0, download=False 2021-11-07 13:09:29 -06:00
Adam J. Stewart f436f04b02
Fix tutorial docs 2021-11-07 13:08:55 -06:00
Adam J. Stewart e22cf22046
Specify test data dir in config 2021-11-07 11:32:02 -06:00
Adam J. Stewart 1a9de994a7
Update installation instructions 2021-11-07 11:09:41 -06:00
Adam J. Stewart 4e708a57fe
Fix local docs build 2021-11-07 11:03:43 -06:00
Adam J. Stewart aa8b6b2dfc
Test more trainers 2021-11-07 10:52:20 -06:00
Adam J. Stewart 0c4e6b517a
setup.py should not be run directly 2021-11-07 10:46:46 -06:00
Adam J. Stewart d5104004c7
Update development status 2021-11-07 10:46:09 -06:00
Adam J. Stewart 363df2747b
Train deps needed for release testing 2021-11-07 10:45:49 -06:00
Adam J. Stewart ab41d56af6
0.1.0 release 2021-11-07 10:28:59 -06:00
Caleb Robinson e5bbc738a3
Adding RESISC45 trainer with augmentations (#225) 
* Removing some keys from the defaults that were removed in Lightning 1.5

* Adding RESISC45

* Experimenting with different LR decay

* Formatting

* Updating RESISC45 test data to be the same size as the original data

* RESISC45 trainer and tests

* Fix import, add deprecation

* mypy fixes

Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
 2021-11-07 05:17:57 +00:00
Caleb Robinson b87d2707ae
Adding the SeCo patch datasets (#223) 
* Adding the SeCo patch datasets

* Adding tests and incorporating suggestions

* Added benchmark example code link

* Update to new way of downloading

* Formatting

* Test coverage

* Rename dataset

* Add SeCo dataset to docs
 2021-11-06 23:58:36 -05:00
Adam J. Stewart 967b4b171f
Trainers: split tasks into separate files, add SemanticSegmentationTask (#224) 
* Trainers: split tasks into separate files

* Add SemanticSegmentationTask

* Fix doc tests

* Keep dataset-specific tasks in separate files

* Remove duplicate So2Sat trainer
 2021-11-06 23:57:40 -05:00
isaac 4257aec131
BigEarthNet Splits (#221) 
* adds split files, new classes, and refactors to speed up loading

* update bigearthnetdatamodule and tests

* update configs

* merge

* force add split files

* fix docstring

* fix docstrings x2

* fix docstrings x3

* Reduce total testing time

Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
 2021-11-05 16:58:25 +00:00
isaac 187d910afd
Refactor Classification Tasks (#222) 2021-11-05 00:04:11 -05:00
Adam J. Stewart 4f6e2096c8
Move DataModules from torchgeo.trainers to torchgeo.datasets (#220) 2021-11-04 17:15:29 -05:00
Adam J. Stewart ea55703bce
Add uncompressed BigEarthNet test data 2021-11-04 20:48:34 +00:00
Adam J. Stewart aa4cec8077
Add COWC trainer (#217) 2021-11-04 08:12:29 -05:00
Caleb Robinson 60674cc200
Adding timm models to the classification task and refactoring (#210) 
* Adding support for VGG models in the classification task
Refactoring the logic for replacing the first conv layer in a network

* Fix formatting

* Testing the stuff

* mypy with torch is such a waste of time

* Update torchgeo/trainers/utils.py

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

* Update torchgeo/trainers/utils.py

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

* Adding timm dependency

* Incorporate timm into ClassificationTask

* Fix tests?

* Formatting

* Allow for overriding stride and padding in `reinit_initial_conv_layer`

* Putting back some stuff I accidentally overwrote in the rebase

* Bug

* Format

Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
 2021-11-03 15:34:33 -07:00
Caleb Robinson de48bd54ad
Adding splits to RESISC45 and EuroSat (#218) 2021-11-02 22:26:39 -05:00