Skip to content
Permalink
Browse files

enhance comments

  • Loading branch information...
ocourtin committed Mar 15, 2019
1 parent 49ae014 commit 1e485c3f0a2d4c45ada88b3ec9c31dc06cb59f80
Showing with 22 additions and 10 deletions.
  1. +22 −10 config.toml
@@ -8,8 +8,26 @@
pg_dsn = "host=127.0.0.1 dbname=rsp user=postgres"


# Classes configuration.

# Input channels configuration.
# You can, add several channels blocks to compose your input Tensor. Order is meaningful.
#
# sub: dataset subdirectory name
# bands: bands to keep from sub source. Order is meaningful
# mean: bands mean value
# std: bands std value
# Nota: (default mean and std are based on ImageNet DataSet, cf pretrained model)
[[channels]]
sub = "images"
bands = [1, 2, 3]
mean = [0.485, 0.456, 0.406]
std = [0.229, 0.224, 0.225]



# Output Classes configuration.
# Nota: available colors are either CSS3 colors names or #RRGGBB hexadecimal representation.
# Nota: only support binary classification for now.
[[classes]]
title = "background"
color = "white"
@@ -19,14 +37,6 @@
color = "deeppink"


# Indicate which dataset sub-directory and bands to take as input.
# You could so, add several channels blocks to compose your input Tensor. Orders are meaningful.
[[channels]]
sub = "images"
bands = [1, 2, 3]
mean = [0.485, 0.456, 0.406]
std = [0.229, 0.224, 0.225]


[model]
# Model name.
@@ -40,8 +50,9 @@

# Loss function name.
loss = "lovasz"

# Batch size for training.
# Nota: can be increase upon your available GPU RAM.
batch_size = 2

# tile side size in pixels.
@@ -51,6 +62,7 @@
epochs = 10

# Learning rate for the optimizer.
# NOTA: should be increase to ~0.0001 if you're not using pretrained models.
lr = 0.000025

# Data augmentation, Flip or Rotate probability.

0 comments on commit 1e485c3

Please sign in to comment.
You can’t perform that action at this time.