search

ceccocats / tkDNN

Deep neural network library and toolkit to do high performace inference on NVIDIA jetson platforms
GNU General Public License v2.0
718 stars 209 forks source link

Do you have any plan to support EfficientDet series? #160

Open TPFRL opened 3 years ago

TPFRL commented 3 years ago

Thanks for making this awesome project!

It will be really great if EfficientDet-series are available on tkDNN platform. As I understand, three additional features need to be implemented:

  1. swish activation function
  2. [scale_channels] layer
  3. [shortcut] layer with weighted multi input

Do you have any plan to support any of these features?

mive93 commented 3 years ago

Hi @TPFRL Thank you! For now, we do not have in plan to add them, however it could be possible that in the future we'll support them. Generally, we port the thing that we need for the projects we work on.

In any case, thank you for your notes! I will let you know if we'll start working on that.

chjej202 commented 3 years ago

Hi @mive93

Somebody created a new pull request (#172) which implements scale_channels and swish activation to support enet-coco.cfg

How about merge this pull request to your branch?

mive93 commented 3 years ago

Hi @chjej202 we have many things to merge to the master (segmentation networks, 3d object detection, tracking), but we have several things to do first at work. As soon as we'll have time, we'll consider that also. Thanks for the remainder in any case.