Skip to content

GitLab

"examples/git@developer.sourcefind.cn:OpenDAS/ollama.git" did not exist on "ad88799411f2e928e09085ec8885fc84c2e9e4f1"
Commit 85b64893 authored by Mikhail Erofeev's avatar Mikhail Erofeev Committed by Yukun Zhu
Browse files

Fix DeepLab Python 3 compatibility issue #6797. (#6799)

parent 0b09300c
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 6 deletions
research/deeplab/eval.py
View file @ 85b64893
...@@ -88,7 +88,7 @@ def main(unused_argv): ...@@ -88,7 +88,7 @@ def main(unused_argv):
split_name=FLAGS.eval_split, split_name=FLAGS.eval_split,
dataset_dir=FLAGS.dataset_dir, dataset_dir=FLAGS.dataset_dir,
batch_size=FLAGS.eval_batch_size, batch_size=FLAGS.eval_batch_size,
crop_size=map(int, FLAGS.eval_crop_size), crop_size=[int(sz) for sz in FLAGS.eval_crop_size],
min_resize_value=FLAGS.min_resize_value, min_resize_value=FLAGS.min_resize_value,
max_resize_value=FLAGS.max_resize_value, max_resize_value=FLAGS.max_resize_value,
resize_factor=FLAGS.resize_factor, resize_factor=FLAGS.resize_factor,
...@@ -106,7 +106,7 @@ def main(unused_argv): ...@@ -106,7 +106,7 @@ def main(unused_argv):
model_options = common.ModelOptions( model_options = common.ModelOptions(
outputs_to_num_classes={common.OUTPUT_TYPE: dataset.num_of_classes}, outputs_to_num_classes={common.OUTPUT_TYPE: dataset.num_of_classes},
crop_size=map(int, FLAGS.eval_crop_size), crop_size=[int(sz) for sz in FLAGS.eval_crop_size],
atrous_rates=FLAGS.atrous_rates, atrous_rates=FLAGS.atrous_rates,
output_stride=FLAGS.output_stride) output_stride=FLAGS.output_stride)
......
research/deeplab/train.py
View file @ 85b64893
...@@ -213,7 +213,7 @@ def _build_deeplab(iterator, outputs_to_num_classes, ignore_label): ...@@ -213,7 +213,7 @@ def _build_deeplab(iterator, outputs_to_num_classes, ignore_label):
model_options = common.ModelOptions( model_options = common.ModelOptions(
outputs_to_num_classes=outputs_to_num_classes, outputs_to_num_classes=outputs_to_num_classes,
crop_size=map(int, FLAGS.train_crop_size), crop_size=[int(sz) for sz in FLAGS.train_crop_size],
atrous_rates=FLAGS.atrous_rates, atrous_rates=FLAGS.atrous_rates,
output_stride=FLAGS.output_stride) output_stride=FLAGS.output_stride)
...@@ -446,7 +446,7 @@ def main(unused_argv): ...@@ -446,7 +446,7 @@ def main(unused_argv):
split_name=FLAGS.train_split, split_name=FLAGS.train_split,
dataset_dir=FLAGS.dataset_dir, dataset_dir=FLAGS.dataset_dir,
batch_size=clone_batch_size, batch_size=clone_batch_size,
crop_size=map(int, FLAGS.train_crop_size), crop_size=[int(sz) for sz in FLAGS.train_crop_size],
min_resize_value=FLAGS.min_resize_value, min_resize_value=FLAGS.min_resize_value,
max_resize_value=FLAGS.max_resize_value, max_resize_value=FLAGS.max_resize_value,
resize_factor=FLAGS.resize_factor, resize_factor=FLAGS.resize_factor,
......
research/deeplab/vis.py
View file @ 85b64893
...@@ -193,7 +193,7 @@ def main(unused_argv): ...@@ -193,7 +193,7 @@ def main(unused_argv):
split_name=FLAGS.vis_split, split_name=FLAGS.vis_split,
dataset_dir=FLAGS.dataset_dir, dataset_dir=FLAGS.dataset_dir,
batch_size=FLAGS.vis_batch_size, batch_size=FLAGS.vis_batch_size,
crop_size=map(int, FLAGS.vis_crop_size), crop_size=[int(sz) for sz in FLAGS.vis_crop_size],
min_resize_value=FLAGS.min_resize_value, min_resize_value=FLAGS.min_resize_value,
max_resize_value=FLAGS.max_resize_value, max_resize_value=FLAGS.max_resize_value,
resize_factor=FLAGS.resize_factor, resize_factor=FLAGS.resize_factor,
...@@ -222,7 +222,7 @@ def main(unused_argv): ...@@ -222,7 +222,7 @@ def main(unused_argv):
model_options = common.ModelOptions( model_options = common.ModelOptions(
outputs_to_num_classes={common.OUTPUT_TYPE: dataset.num_of_classes}, outputs_to_num_classes={common.OUTPUT_TYPE: dataset.num_of_classes},
crop_size=map(int, FLAGS.vis_crop_size), crop_size=[int(sz) for sz in FLAGS.vis_crop_size],
atrous_rates=FLAGS.atrous_rates, atrous_rates=FLAGS.atrous_rates,
output_stride=FLAGS.output_stride) output_stride=FLAGS.output_stride)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment