tensorflow / tensorflow/models
Pre-trained CenterNet with MobileNet v2 FPN produces no detections
Open
@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
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
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.