Skip to content

Commit 7de7995

Browse files
update nnom stuff
1 parent 059448f commit 7de7995

File tree

4 files changed

+12
-4
lines changed

4 files changed

+12
-4
lines changed

docker_compilers/nnom/v0.4.3/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
FROM tensorflow/tensorflow:2.14.0rc1
1+
FROM tensorflow/tensorflow:2.14.0
22
LABEL Description="Image for generating NNOM from model file"
33

44

src/embedded_ml_sdk/classifiers/nnom_middleware/nnom_middleware.c

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,15 +35,20 @@ uint8_t nnom_simple_submit(uint8_t classifier_id, feature_vector_t *feature_vect
3535
{
3636
uint8_t y = 1;
3737
float max_result = 0;
38+
feature_vector_data = (int8_t*)feature_vector->data;
3839

39-
if (last_nnom_initialized != classifier_id)
40+
/*if (last_nnom_initialized != classifier_id)
4041
{
4142
model = nnom_model_create();
4243
//nnom_classifier_rows[0].model=model
4344
last_nnom_initialized = classifier_id;
4445
}
46+
*/
47+
48+
for (int i=0; i<feature_vector->size; i++){
4549

46-
memcpy(nnom_input_data, feature_vector->data, feature_vector->size);
50+
nnom_input_data[i]=feature_vector_data[i]
51+
}
4752
nnom_predict(nnom_classifier_rows[classifier_id].model, &label, &probability);
4853

4954

src/server/codegen/templates/compilers/nnom/build.sh

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,9 @@ mkdir libsensiml/nnom
115115
cp -r /nnom/inc/* libsensiml/nnom/
116116
cp /nnom/port/* libsensiml/nnom/
117117
cp /nnom/src/core/* libsensiml/nnom/
118-
cp -r /nnom/src/layers/* libsensiml/nnom/layers/
118+
cp -r /nnom/src/layers/*.c libsensiml/nnom/
119+
mkdir libsensiml/nnom/layers
120+
cp -r /nnom/src/layers/*.h libsensiml/nnom/layers/
119121

120122
cp /nnom/src/backends/* libsensiml/nnom/
121123
cp nnom/weights.h libsensiml/nnom/

src/server/engine/recognitionengine.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -435,6 +435,7 @@ def reco_kb_pipeline(self, kb_description):
435435
"target_compiler": CompilerDescription.objects.get(
436436
uuid="62aabe7e-4f5d-4167-a786-072e4a8dc158"
437437
),
438+
"nn_inference_engine":'nnom'
438439
}
439440

440441
logger.userlog(

0 commit comments

Comments
 (0)