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
17514064
Commit
17514064
authored
Jan 05, 2017
by
Neal Wu
Committed by
GitHub
Jan 05, 2017
Browse files
Merge pull request #834 from BoyuanJiang/patch-2
Update build_image_data.py
parents
bfa1bb1b
9997b250
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
5 deletions
+5
-5
inception/inception/data/build_image_data.py
inception/inception/data/build_image_data.py
+5
-5
No files found.
inception/inception/data/build_image_data.py
View file @
17514064
...
...
@@ -137,13 +137,13 @@ def _convert_to_example(filename, image_buffer, label, text, height, width):
example
=
tf
.
train
.
Example
(
features
=
tf
.
train
.
Features
(
feature
=
{
'image/height'
:
_int64_feature
(
height
),
'image/width'
:
_int64_feature
(
width
),
'image/colorspace'
:
_bytes_feature
(
colorspace
),
'image/colorspace'
:
_bytes_feature
(
tf
.
compat
.
as_bytes
(
colorspace
)
)
,
'image/channels'
:
_int64_feature
(
channels
),
'image/class/label'
:
_int64_feature
(
label
),
'image/class/text'
:
_bytes_feature
(
text
),
'image/format'
:
_bytes_feature
(
image_format
),
'image/filename'
:
_bytes_feature
(
os
.
path
.
basename
(
filename
)),
'image/encoded'
:
_bytes_feature
(
image_buffer
)}))
'image/class/text'
:
_bytes_feature
(
tf
.
compat
.
as_bytes
(
text
)
)
,
'image/format'
:
_bytes_feature
(
tf
.
compat
.
as_bytes
(
image_format
)
)
,
'image/filename'
:
_bytes_feature
(
tf
.
compat
.
as_bytes
(
os
.
path
.
basename
(
filename
))
)
,
'image/encoded'
:
_bytes_feature
(
tf
.
compat
.
as_bytes
(
image_buffer
)
)
}))
return
example
...
...
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