added check to make sure not to return null feature maps

kylemcdonald · Aug 30, 2016 · 7f23db5 · 7f23db5
1 parent f4eca25
commit 7f23db5
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/ofxCcv.cpp b/src/ofxCcv.cpp
@@ -102,11 +102,13 @@ void ofxCcv::FeatureMap::getImage(ofImage & img, bool autoBrighten) {
 }
 
 vector<ofxCcv::FeatureMap> ofxCcv::getFeatureMaps(int layer) const {
+	vector<FeatureMap> maps;
+    if (convnet->acts[layer] == NULL) {
+        return maps;
+    }
     int rows = convnet->acts[layer]->rows;
     int cols = convnet->acts[layer]->cols;
     int channels = CCV_GET_CHANNEL(convnet->acts[layer]->type);
-
-    vector<FeatureMap> maps;
     for (int c=0; c<channels; c++) {
         FeatureMap map;
         map.rows = rows;