Table 1.
Hyperparameter | Value |
---|---|
IMAGES_PER_GPU | 4 |
GPU_COUNT | 4 |
STEPS_PER_EPOCH | 10 |
IMAGE_MAX_DIM | 448 |
IMAGE_MIN_DIM | 448 |
LAYER_1 | 60 |
LAYER_2 | 120 |
LAYER_3 | 200 |
DETECTION_MIN_CONFIDENCE | 0.9 |
LEARNING_RATE | 0.001 |
LEARNING_MOMENTUM | 0.9 |
WEIGHT_DECAY | 0.0001 |
DETECTION_MAX_INSTANCES | 100 |