This commit is contained in:
Caleb Robinson 2021-08-31 17:58:08 +00:00 коммит произвёл Adam J. Stewart
Родитель 174ddfc15d
Коммит 3e296991e2
17 изменённых файлов: 41 добавлений и 45 удалений

Просмотреть файл

@ -1,6 +1,3 @@
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
[flake8]
max-line-length = 88
extend-ignore =

3
.gitattributes поставляемый
Просмотреть файл

@ -1,5 +1,2 @@
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
# Do not change line endings on test data, it will change the MD5
/tests/data/*/** binary

3
.gitignore поставляемый
Просмотреть файл

@ -1,6 +1,3 @@
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
# TorchGeo
/data/
/logs/

Просмотреть файл

@ -1,6 +1,3 @@
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
program: # These are experiment level arguments
experiment_name: cyclone_test
overwrite: True

Просмотреть файл

@ -1,6 +1,3 @@
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
config_file: null # The user can pass a filename here on the command line
program: # These are the default arguments

Просмотреть файл

@ -0,0 +1,17 @@
program:
experiment_name: cyclone_resnet18_baseline
overwrite: True
data_dir: /home/caleb/mount/data/cyclone
output_dir: output/
log_dir: output/logs/
batch_size: 64
trainer:
min_epochs: 30
gpus: 1
task:
name: "cyclone"
learning_rate: 1e-3
learning_rate_schedule_patience: 2
model: "resnet18"

Просмотреть файл

@ -0,0 +1,24 @@
program: # These are experiment level arguments
experiment_name: landcoverai_unet_resnet18_imagenet_fast
overwrite: True
data_dir: /home/caleb/mount/data
output_dir: output/
log_dir: output/logs/
batch_size: 32
trainer: # These are all the arguments that will be passed to the pl.Trainer
gpus: 1
min_epochs: 40
max_epochs: 100
benchmark: True
task: # These are all the arguments that will be used to create an appropriate task
name: landcoverai
optimizer: "adamw"
learning_rate: 1e-2
learning_rate_schedule_patience: 7
loss: "jaccard"
segmentation_model: "unet"
encoder_name: "resnet18"
encoder_weights: "imagenet"
encoder_output_stride: 16

Просмотреть файл

@ -1,6 +1,3 @@
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
program: # These are experiment level arguments
experiment_name: landcoverai_test
overwrite: True

Просмотреть файл

@ -1,6 +1,3 @@
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
program: # These are experiment level arguments
experiment_name: sen12ms_test
overwrite: True

Просмотреть файл

@ -1,6 +1,3 @@
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
task:
name: "cyclone"
learning_rate: 1e-3

Просмотреть файл

@ -1,6 +1,3 @@
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
task:
name: "landcoverai"
optimizer: "adamw"

Просмотреть файл

@ -1,6 +1,3 @@
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
task:
name: "sen12ms"
learning_rate: 1e-3

Просмотреть файл

@ -1,6 +1,3 @@
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
name: torchgeo
channels:
- conda-forge

Просмотреть файл

@ -1,6 +1,3 @@
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
[build-system]
requires = [
"setuptools>=42",

Просмотреть файл

@ -1,6 +1,3 @@
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
# black 21+ required for Python 3.9 support
black>=21.4b0
# fiona 1.5+ required for fiona.transform module

Просмотреть файл

@ -1,6 +1,3 @@
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
# https://setuptools.readthedocs.io/en/latest/userguide/declarative_config.html
[metadata]
name = torchgeo

Просмотреть файл

@ -1,6 +1,3 @@
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
spack:
specs:
- "opencv+python3+imgcodecs+tiff+jpeg+png"