tensorflow / models

Models and examples built with TensorFlow
Other
77.03k stars 45.77k forks source link

Any plans to release EfficientNet based SSD for object detection ? [Question] #6996

Open dexception opened 5 years ago

dexception commented 5 years ago

Any plans to release EfficientNet based SSD for object detection ?

tensorflowbutler commented 5 years ago

Thank you for your post. We noticed you have not filled out the following field in the issue template. Could you update them if they are relevant in your case, or leave them as N/A? Thanks. What is the top-level directory of the model you are using Have I written custom code OS Platform and Distribution TensorFlow installed from TensorFlow version Bazel version CUDA/cuDNN version GPU model and memory Exact command to reproduce

dexception commented 5 years ago

What is the top-level directory of the model you are using : N/A Have I written custom code: No OS Platform and Distribution: Ubuntu 16.04 TensorFlow installed from: Pip TensorFlow version: 1.13.1 Bazel version: N/A CUDA/cuDNN version: 9.0 GPU model and memory: 1080 TI Exact command to reproduce: N/A

lukasschmit commented 5 years ago

I'm also very interested in this. I'm currently using the tpu implementations of retinanet and efficientnet together, but that code is very centered on tpu training and I've had to rewrite a bunch just to make it work on GPUs. Plus, the TFRecord input format is a pain file-size-wise. Would be great to get an implementation with better support for people who can't afford TPUs.

PythonImageDeveloper commented 5 years ago

Hi @lukasschmit How do you train the model on tpu device? Do you use the google colab service? When the model trained using tpu device, Can be extract the freeze_graph file (.pb) using the GPU device or is necessary only tpu device and tpu support python file for this work?

dexception commented 4 years ago

This looks like a complete unprofessional project.

No Imagenet models for the following:

  1. ResneXt
  2. EfficientNet B0-B7
  3. ShuffleNet
  4. Mixnet S-M-L
  5. MobileNetV3
  6. FbNet
  7. SENet
  8. GhostNet : https://arxiv.org/abs/1911.11907
noahtren commented 4 years ago

There are TF2 (Keras) compatible ImageNet models for EfficientNet, just not part of this repo yet. See https://github.com/keras-team/keras-applications

Regarding EfficientNet for object detection, you might be interested in EfficientDet which is a recent variation of EfficientNet for object detection. This is a very recent paper and the code isn't public yet. https://arxiv.org/abs/1911.09070

There's an issue on tensorflow/tpu tracking when it will be released https://github.com/tensorflow/tpu/issues/611

dexception commented 4 years ago

I have to ask... why there has been absolutely no work in this repo in last few months ?

dynamicwebpaige commented 4 years ago

A mobile-compatible EfficientNet was released this month. Please try it out, and make sure to check the release notes for specific features. Thanks!

bhack commented 4 years ago

An anchor free version with EfficientDet backbone is available at https://github.com/xuannianz/SAPD.

For others anchor free models https://github.com/tensorflow/hub/issues/424

@dynamicwebpaige P.s. Users still need to always spread tickets everywhere https://github.com/tensorflow/tpu/issues/611. I hope that we could centralize models.

bhack commented 4 years ago

See also https://github.com/tensorflow/tpu/issues/696 https://github.com/tensorflow/tpu/issues/664

bhack commented 4 years ago

And I suppose we need also an EfficientDet-lite version after https://blog.tensorflow.org/2020/03/higher-accuracy-on-vision-models-with-efficientnet-lite.html

bhack commented 4 years ago

The release was announced today but in another Google (AutoML) repo.. just to have a little bit more fragmentation. See https://github.com/tensorflow/tpu/issues/611#issuecomment-600192115

saberkun commented 4 years ago

Hi, EfficientDet is open sourced by Mingxing in automl dedicated repository. Official code which is TF1 implement is here: https://github.com/google/automl/tree/master/efficientdet

We are very close to release EfficientNet image classification training scripts (currently for basic EfficientNet architecture) in tensorflow/models/official/vision.

bhack commented 4 years ago

@saberkun I hope that from a pratical point of view we could use It also with new Efficientnet light backbone. /cc @mingxingtan I think that having some example in TFHUB with anchor-free solutions It could simplify the pipeline for many in object keypoits/landmarks task (face landmarks, pose keypoints, in object landmarks, etc..).