Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
afb2a7dc
Unverified
Commit
afb2a7dc
authored
May 18, 2018
by
Yukun Zhu
Committed by
GitHub
May 18, 2018
Browse files
Update ade20k.md
parent
20d75ea0
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
10 deletions
+3
-10
research/deeplab/g3doc/ade20k.md
research/deeplab/g3doc/ade20k.md
+3
-10
No files found.
research/deeplab/g3doc/ade20k.md
View file @
afb2a7dc
...
...
@@ -49,7 +49,7 @@ A local training job using `xception_65` can be run with the following command:
# From tensorflow/models/research/
python deeplab/train.py
\
--logtostderr
\
--training_number_of_steps
=
5
0000
\
--training_number_of_steps
=
9
0000
\
--train_split
=
"train"
\
--model_variant
=
"xception_65"
\
--atrous_rates
=
6
\
...
...
@@ -60,21 +60,16 @@ python deeplab/train.py \
--train_crop_size
=
513
\
--train_crop_size
=
513
\
--train_batch_size
=
4
\
--min_resize_value
=
350
\
--max_resize_value
=
5
00
\
--min_resize_value
=
513
\
--max_resize_value
=
5
13
\
--resize_factor
=
16
\
--fine_tune_batch_norm
=
False
\
--dataset
=
"ade20k"
\
--initialize_last_layer
=
False
\
--last_layers_contain_logits_only
=
True
\
--tf_initial_checkpoint
=
${
PATH_TO_INITIAL_CHECKPOINT
}
\
--train_logdir
=
${
PATH_TO_TRAIN_DIR
}
\
--dataset_dir
=
${
PATH_TO_DATASET
}
```
where ${PATH
\_
TO
\_
INITIAL
\_
CHECKPOINT} is the path to the initial checkpoint.
For example, if you are using the deeplabv3
\_
pascal
\_
train
\_
aug checkppoint, you
will set this to
`/path/to/deeplabv3\_pascal\_train\_aug/model.ckpt`
.
${PATH
\_
TO
\_
TRAIN
\_
DIR} is the directory in which training checkpoints and
events will be written to (it is recommended to set it to the
`train_on_train_set/train`
above), and ${PATH
\_
TO
\_
DATASET} is the directory in
...
...
@@ -98,8 +93,6 @@ which the ADE20K dataset resides (the `tfrecord` above)
4.
The users could skip the flag,
`decoder_output_stride`
, if you do not want
to use the decoder structure.
Currently there are no fine-tuned checkpoint for the ADE20K dataset.
## Running Tensorboard
Progress for training and evaluation jobs can be inspected using Tensorboard. If
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment