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