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
545b75f9
Commit
545b75f9
authored
Jul 06, 2017
by
Thibaut Mattio
Browse files
Remove six dependency for dictionnaries iterators
parent
ff64c849
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
2 deletions
+1
-2
object_detection/utils/visualization_utils.py
object_detection/utils/visualization_utils.py
+1
-2
No files found.
object_detection/utils/visualization_utils.py
View file @
545b75f9
...
...
@@ -25,7 +25,6 @@ import PIL.Image as Image
import
PIL.ImageColor
as
ImageColor
import
PIL.ImageDraw
as
ImageDraw
import
PIL.ImageFont
as
ImageFont
import
six
import
tensorflow
as
tf
...
...
@@ -398,7 +397,7 @@ def visualize_boxes_and_labels_on_image_array(image,
classes
[
i
]
%
len
(
STANDARD_COLORS
)]
# Draw all boxes onto image.
for
box
,
color
in
six
.
iteritems
(
box_to_color_map
):
for
box
,
color
in
box_to_color_map
.
items
(
):
ymin
,
xmin
,
ymax
,
xmax
=
box
if
instance_masks
is
not
None
:
draw_mask_on_image_array
(
...
...
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