Spaces:
Sleeping
Sleeping
# Written by Dr Daniel Buscombe, Marda Science LLC | |
# for the SandSnap Program | |
# | |
# MIT License | |
# | |
# Copyright (c) 2020-2021, Marda Science LLC | |
# | |
# Permission is hereby granted, free of charge, to any person obtaining a copy | |
# of this software and associated documentation files (the "Software"), to deal | |
# in the Software without restriction, including without limitation the rights | |
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | |
# copies of the Software, and to permit persons to whom the Software is | |
# furnished to do so, subject to the following conditions: | |
# | |
# The above copyright notice and this permission notice shall be included in all | |
# copies or substantial portions of the Software. | |
# | |
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | |
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE | |
# SOFTWARE. | |
##> Release v1.4 (Aug 2021) | |
## Contains values for defaults that you may change. | |
## They are listed in order of likelihood that you might change them: | |
# size of image in pixels. keep this consistent in training and application | |
# suggestd: 512 -- 1024 (larger = larger GPU required) | |
# integer | |
IM_HEIGHT = 1024 | |
IM_WIDTH = IM_HEIGHT | |
# number of images to feed the network per step in epoch #suggested: as many as you have gpu memory for, probably | |
# integer | |
# BATCH_SIZE =8 | |
# BATCH_SIZE =10 | |
BATCH_SIZE =12 | |
#use an ensemble of batch sizes like this | |
#BATCH_SIZE = [7,12,14] | |
# if True, use a smaller (shallower) network architecture | |
##True or False ##False=larger network | |
SHALLOW = False #True | |
## if True, carry out data augmentation. 2 x number of images used in training | |
##True or False | |
DO_AUG = False #True | |
# maximum learning rate ##1e-1 -- 1e-5 | |
MAX_LR = 1e-4 | |
# MAX_LR = 1e-5 | |
# MAX_LR = 5e-3 | |
# MAX_LR = 5e-4 | |
# max. number of training epics (20 -1000) | |
# integer | |
NUM_EPOCHS = 300 | |
## loss function for continuous models (2 choices) | |
#CONT_LOSS = 'pinball' | |
CONT_LOSS = 'mse' | |
## loss function for categorical (disrete) models (2 choices) | |
CAT_LOSS = 'focal' | |
#CAT_LOSS = 'categorical_crossentropy' | |
# optimizer (gradient descent solver) good alternative == 'rmsprop' | |
OPT = 'adam' | |
# base number of conv2d filters in categorical models | |
# integer | |
BASE_CAT = 30 | |
# base number of conv2d filters in continuous models | |
# integer | |
# BASE_CONT = 30 | |
BASE_CONT = 10 | |
# number of Dense units for continuous prediction | |
# integer | |
# CONT_DENSE_UNITS = 3072 | |
CONT_DENSE_UNITS = 2048 | |
# CONT_DENSE_UNITS = 1024 | |
# number of Dense units for categorical prediction | |
# integer | |
CAT_DENSE_UNITS = 128 | |
# set to False if you wish to use cpu (not recommended) | |
##True or False | |
USE_GPU = True | |
## standardize imagery (recommended) | |
DO_STANDARDIZE = True | |
# STOP_PATIENCE = 10 | |
# FACTOR = 0.2 | |
# MIN_DELTA = 0.0001 | |
# MIN_LR = 1e-4 |