1. Enable network state write/read

[ti-machine-learning/ti-machine-learning.git] / src / app / cnn / scene / appCNNSceneGetPatch.c

diff --git a/src/app/cnn/scene/appCNNSceneGetPatch.c b/src/app/cnn/scene/appCNNSceneGetPatch.c
index afcc5ad65783a8405e703ae29dce3daa4119ed91..3e36266d2972da63bcd37a47b12e6c84b159e0ec 100644 (file)
--- a/src/app/cnn/scene/appCNNSceneGetPatch.c
+++ b/src/app/cnn/scene/appCNNSceneGetPatch.c
@@ -62,6 +62,7 @@ int appCNNSceneGetPatch(int imageIdx, int rowIdx, int colIdx, appCNNSceneDataSet
    int           i;
    int           j;
    int           k;
+   int           index;
    int           imageRow;
    int           imageCol;
    int           numRow;
@@ -109,15 +110,20 @@ int appCNNSceneGetPatch(int imageIdx, int rowIdx, int colIdx, appCNNSceneDataSet
       for (j = 0; j < patchSize; j++) {
          imageRow = rowIdx - patchSize/2 + i;
          imageCol = colIdx - patchSize/2 + j;
+
          if (imageRow < 0 || imageRow >= numRow || imageCol < 0 || imageCol >= numCol) {
             for (k = 0; k < numChannel; k++) {
-               patch[j + i*patchSize + k*patchSize*patchSize] = 0.0;
+               index = j + i*patchSize + k*patchSize*patchSize;
+               patch[j + i*patchSize + k*patchSize*patchSize] = 0;
             }
          }
          else {
-            for (k = 0; k < numChannel; k++)
+            for (k = 0; k < numChannel; k++) {
+               index = j + i*patchSize + k*patchSize*patchSize;
                patch[j + i*patchSize + k*patchSize*patchSize] = image.data[imageCol + imageRow*numCol + k*numRow*numCol];
+            }
          }
+
       }
    }