Skip to content

Instantly share code, notes, and snippets.

@chricke
Last active July 18, 2018 20:56
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save chricke/b899b7debcfb2fae9ebab545cc8b9722 to your computer and use it in GitHub Desktop.
Save chricke/b899b7debcfb2fae9ebab545cc8b9722 to your computer and use it in GitHub Desktop.
Configuration for training luminoth network for table detection
train:
# Name used to identify the run. Data inside `job_dir` will be stored under
# `run_name`.
run_name: table-area-detection-0.1
# Base directory in which model checkpoints & summaries (for Tensorboard) will
# be saved.
job_dir: jobs/
save_checkpoint_secs: 10
save_summaries_secs: 10
# Number of epochs (complete dataset batches) to run.
num_epochs: 10
dataset:
type: object_detection
# From which directory to read the dataset.
dir: tfdata/classes-table/
image_preprocessing:
min_size: 600
max_size: 1024
data_augmentation:
- flip:
left_right: True
up_down: True
prob: 0.5
model:
type: fasterrcnn
network:
# Total number of classes to predict.
num_classes: 1
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment