RidgeRun / gst-inference

A GStreamer Deep Learning Inference Framework
GNU Lesser General Public License v2.1
123 stars 29 forks source link

TensorFlow Implementation (load mode & expose out param) #9

Closed carlos-aguero closed 4 years ago

carlos-aguero commented 5 years ago

@michaelgruner will pass pointer for download to @GFallasRR - Nov 26

carlos-aguero commented 5 years ago

Make R2I classes: Frame, Prediction, Loader, Factory, Model, Engine, Parameters (Expose parameters: output name, input names).

Acceptance Criteria:

  1. Run TensorFlow working with GoogleNet
carlos-aguero commented 5 years ago

model running now, but not valid output... Michael converted module to NCSDK and output is acceptable now, Michael will check for a new model and validate!

Making code at .c for testing and understanding and after running and validating will create required classes

carlos-aguero commented 5 years ago

Model runs at C, init image normalization was causing problems, people is detected now. Next expose with classes

carlos-aguero commented 5 years ago

Classes (Ready) factory, model, loader Classes (TODO) Frame, Prediction, Engine, Parameters (Expose parameters: output name, input names).

carlos-aguero commented 5 years ago

Unit tests for all classes (except frame and prediction)

carlos-aguero commented 5 years ago
carlos-aguero commented 5 years ago

Tested with valgrind, no leaks reported

carlos-aguero commented 5 years ago

tests are:

  1. loader DO IT
  2. frame DO IT prediction On Hold engine On Hold model On Hold parameters On Hold