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
c96825a4
Commit
c96825a4
authored
Nov 01, 2020
by
Vishnu Banna
Browse files
Merge branch 'yolo' of
https://github.com/PurdueCAM2Project/tf-models
into yolo
parents
d791ef44
9d58e701
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
official/vision/beta/tasks/image_classification.py
official/vision/beta/tasks/image_classification.py
+2
-2
No files found.
official/vision/beta/tasks/image_classification.py
View file @
c96825a4
...
...
@@ -54,10 +54,10 @@ class ImageClassificationTask(base_task.Task):
input_size
=
self
.
task_config
.
model
.
input_size
if
params
.
tfds_name
!=
None
:
print
(
"i am here for training using tfds"
)
tf
.
print
(
"i am here for training using tfds"
)
decoder
=
cli
.
Decoder
()
else
:
print
(
"i am here for regular input"
)
tf
.
print
(
"i am here for regular input"
)
decoder
=
classification_input
.
Decoder
()
parser
=
classification_input
.
Parser
(
...
...
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