"...resnet50_tensorflow.git" did not exist on "7fb4f3cdfa94fabf63bc9f4fa13408be8f9a08df"
Commit db70cac3 authored by Vishnu Banna's avatar Vishnu Banna
Browse files

Merge branch 'detection_generator_pr' of...

Merge branch 'detection_generator_pr' of https://github.com/PurdueCAM2Project/tf-models into detection_generator_pr
parents fd28870e 40f8439d
...@@ -667,7 +667,9 @@ def build_darknet( ...@@ -667,7 +667,9 @@ def build_darknet(
l2_regularizer: tf.keras.regularizers.Regularizer = None) -> tf.keras.Model: l2_regularizer: tf.keras.regularizers.Regularizer = None) -> tf.keras.Model:
"""Builds darknet.""" """Builds darknet."""
backbone_cfg = backbone_config.get() backbone_cfg = model_config.backbone.get()
norm_activation_config = model_config.norm_activation
model = Darknet( model = Darknet(
model_id=backbone_cfg.model_id, model_id=backbone_cfg.model_id,
min_level=backbone_cfg.min_level, min_level=backbone_cfg.min_level,
......
...@@ -14,7 +14,6 @@ ...@@ -14,7 +14,6 @@
# Lint as: python3 # Lint as: python3
"""Contains common building blocks for yolo neural networks.""" """Contains common building blocks for yolo neural networks."""
from typing import Callable, List from typing import Callable, List
import tensorflow as tf import tensorflow as tf
from official.modeling import tf_utils from official.modeling import tf_utils
...@@ -805,7 +804,6 @@ class CSPStack(tf.keras.layers.Layer): ...@@ -805,7 +804,6 @@ class CSPStack(tf.keras.layers.Layer):
"""CSPStack layer initializer. """CSPStack layer initializer.
Args: Args:
filters: integer for output depth, or the number of features to learn.
model_to_wrap: callable Model or a list of callable objects that will model_to_wrap: callable Model or a list of callable objects that will
process the output of CSPRoute, and be input into CSPConnect. process the output of CSPRoute, and be input into CSPConnect.
list will be called sequentially. list will be called sequentially.
......
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