tensorflow / tensorflow/models

Pre-trained CenterNet with MobileNet v2 FPN produces no detections

Open
#10,506 7 comments 0 reactions 3 assignees View on GitHub

@pkulzc is already working on this.

Since Feb 22, 2022.

models:research:odapi type:bug
Dominant language
Python
Stars
77.7k
Forks
44.8k
PR merge metrics
No merged PRs in 30d

Description

Prerequisites

Please answer the following questions for yourself before submitting an issue.

  • I am using the latest TensorFlow Model Garden release and TensorFlow 2.
  • I am reporting the issue to the correct repository. (Model Garden official or research directory)
  • I checked to make sure that this issue has not already been filed.

1. The entire URL of the file you are using

http://download.tensorflow.org/models/object_detection/tf2/20210210/centernet_mobilenetv2fpn_512x512_coco17_od.tar.gz

2. Describe the bug

Running https://github.com/tensorflow/models/blob/master/research/object_detection/colab_tutorials/inference_tf2_colab.ipynb (modified to not be a python notebook and to use the referenced model) only generates detections with a score <= 1e-7

3. Steps to reproduce

Follow the steps described in the notebook.

4. Expected behavior

At least 1 detection with a "positive" score

5. Additional context

The output from running the script

WARNING:tensorflow:`input_shape` is undefined or non-square, or `rows` is not in [96, 128, 160, 192, 224]. Weights for input shape (224, 224) will be loaded as the default.
2022-02-17 16:38:37.502211: I tensorflow/core/platform/cpu_feature_guard.cc:151] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations:  AVX2 FMA
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
2022-02-17 16:38:39.079733: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1525] Created device /job:localhost/replica:0/task:0/device:GPU:0 with 8829 MB memory:  -> device: 0, name: NVIDIA GeForce GTX 1080 Ti, pci bus id: 0000:09:00.0, compute capability: 6.1
2022-02-17 16:38:39.080309: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1525] Created device /job:localhost/replica:0/task:0/device:GPU:1 with 10407 MB memory:  -> device: 1, name: NVIDIA GeForce GTX 1080 Ti, pci bus id: 0000:0a:00.0, compute capability: 6.1
2022-02-17 16:38:39.080797: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1525] Created device /job:localhost/replica:0/task:0/device:GPU:2 with 10405 MB memory:  -> device: 2, name: NVIDIA GeForce GTX 1080 Ti, pci bus id: 0000:43:00.0, compute capability: 6.1
2022-02-17 16:38:47.909083: I tensorflow/stream_executor/cuda/cuda_dnn.cc:368] Loaded cuDNN version 8301                                                                                                                                                                       tf.Tensor(
[[2.6671066e-07 2.3892198e-07 2.3317602e-07 1.7589170e-07 9.5514459e-08
  7.8194894e-08 7.0393163e-08 6.9936938e-08 6.0256994e-08 3.7475342e-08
  3.2252693e-08 2.8360157e-08 1.5607364e-08 1.2781412e-08 8.6965866e-09
  8.4613019e-09 7.2930542e-09 7.0490271e-09 5.1227773e-09 4.5339954e-09]], shape=(1, 20), dtype=float32)

6. System information

  • OS Platform and Distribution (e.g., Linux Ubuntu 16.04): Arch Linux
  • Mobile device name if the issue happens on a mobile device:
  • TensorFlow installed from (source or binary): Installed from Arch Linux package repository (community/tensorflow-opt-cuda)
  • TensorFlow version (use command below): v2.8.0-rc1-32-g3f878cff5b6 2.8.0
  • Python version: 3.10.2
  • Bazel version (if compiling from source):
  • GCC/Compiler version (if compiling from source):
  • CUDA/cuDNN version: 11.6/8.3
  • GPU model and memory: Nvidia GTX1080Ti 11GB

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.