tensorflow/models

File exporter_lib_v2.py has 267 lines of code (exceeds 250 allowed). Consider refactoring.

Open

# Copyright 2020 The TensorFlow Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at

Function export_inference_graph has 8 arguments (exceeds 4 allowed). Consider refactoring.

Open

def export_inference_graph(input_type,

Similar blocks of code found in 2 locations. Consider refactoring.

Open

class DetectionFromTFExampleModule(DetectionInferenceModule):
  """Detection Inference Module for TF.Example inputs."""

  @tf.function(input_signature=[tf.TensorSpec(shape=[None], dtype=tf.string)])
  def __call__(self, input_tensor):

• Read up
  Read up

Similar blocks of code found in 2 locations. Consider refactoring.

Open

class DetectionFromEncodedImageModule(DetectionInferenceModule):
  """Detection Inference Module for encoded image string inputs."""

  @tf.function(input_signature=[tf.TensorSpec(shape=[None], dtype=tf.string)])
  def __call__(self, input_tensor):

• Read up
  Read up

Identical blocks of code found in 2 locations. Consider refactoring.

Open

    for key, val in detections.items():
      detections[key] = tf.cast(val, tf.float32)

• Read up
  Read up

Identical blocks of code found in 2 locations. Consider refactoring.

Open

    for key, val in detections.items():
      detections[key] = tf.cast(val, tf.float32)

• Read up
  Read up

Identical blocks of code found in 2 locations. Consider refactoring.

Open

    detections[classes_field] = (
        tf.cast(detections[classes_field], tf.float32) + label_id_offset)

• Read up
  Read up

Identical blocks of code found in 2 locations. Consider refactoring.

Open

    detections[classes_field] = (
        tf.cast(detections[classes_field], tf.float32) + label_id_offset)

• Read up
  Read up