Skip to content

Instantly share code, notes, and snippets.

@chricke

chricke/config.yml

Last active Jul 18, 2018
Embed
What would you like to do?
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
You can’t perform that action at this time.