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
350b0aa6
Commit
350b0aa6
authored
Sep 09, 2021
by
Vishnu Banna
Browse files
decoder update
parent
7752c35a
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
4 deletions
+4
-4
official/vision/beta/projects/yolo/modeling/decoders/yolo_decoder.py
...sion/beta/projects/yolo/modeling/decoders/yolo_decoder.py
+4
-4
No files found.
official/vision/beta/projects/yolo/modeling/decoders/yolo_decoder.py
View file @
350b0aa6
...
@@ -13,7 +13,7 @@
...
@@ -13,7 +13,7 @@
# limitations under the License.
# limitations under the License.
# Lint as: python3
# Lint as: python3
"""Feature Pyramid Network and Path Aggregation variants used in YOLO"""
"""Feature Pyramid Network and Path Aggregation variants used in YOLO
.
"""
import
tensorflow
as
tf
import
tensorflow
as
tf
from
official.vision.beta.projects.yolo.modeling.layers
import
nn_blocks
from
official.vision.beta.projects.yolo.modeling.layers
import
nn_blocks
...
@@ -206,7 +206,7 @@ class YoloPAN(tf.keras.layers.Layer):
...
@@ -206,7 +206,7 @@ class YoloPAN(tf.keras.layers.Layer):
by zero.
by zero.
kernel_initializer: kernel_initializer for convolutional layers.
kernel_initializer: kernel_initializer for convolutional layers.
kernel_regularizer: tf.keras.regularizers.Regularizer object for Conv2D.
kernel_regularizer: tf.keras.regularizers.Regularizer object for Conv2D.
bias_regularizer: tf.keras.regularizers.Regularizer object for Conv2
d
.
bias_regularizer: tf.keras.regularizers.Regularizer object for Conv2
D
.
fpn_input: `bool`, for whether the input into this fucntion is an FPN or
fpn_input: `bool`, for whether the input into this fucntion is an FPN or
a backbone.
a backbone.
fpn_filter_scale: `int`, scaling factor for the FPN filters.
fpn_filter_scale: `int`, scaling factor for the FPN filters.
...
@@ -381,7 +381,7 @@ class YoloDecoder(tf.keras.Model):
...
@@ -381,7 +381,7 @@ class YoloDecoder(tf.keras.Model):
"""Yolo Decoder initialization function.
"""Yolo Decoder initialization function.
A unified model that ties all decoder components into a conditionally build
A unified model that ties all decoder components into a conditionally build
YOLO decder.
YOLO dec
o
der.
Args:
Args:
input_specs: `dict[str, tf.InputSpec]`: input specs of each of the inputs
input_specs: `dict[str, tf.InputSpec]`: input specs of each of the inputs
...
@@ -403,7 +403,7 @@ class YoloDecoder(tf.keras.Model):
...
@@ -403,7 +403,7 @@ class YoloDecoder(tf.keras.Model):
zero.
zero.
kernel_initializer: kernel_initializer for convolutional layers.
kernel_initializer: kernel_initializer for convolutional layers.
kernel_regularizer: tf.keras.regularizers.Regularizer object for Conv2D.
kernel_regularizer: tf.keras.regularizers.Regularizer object for Conv2D.
bias_regularizer: tf.keras.regularizers.Regularizer object for Conv2
d
.
bias_regularizer: tf.keras.regularizers.Regularizer object for Conv2
D
.
**kwargs: keyword arguments to be passed.
**kwargs: keyword arguments to be passed.
"""
"""
...
...
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