Update label.py

label.py

@@ -49,7 +49,7 @@ def load_labels():
     return classes, labels_IO, labels_attribute, W_attribute
 
 def hook_feature(module, input, output):
-    features_blobs.append(np.squeeze(output.data.cpu().numpy()))
+    return np.squeeze(output.data.cpu().numpy())
 
 def returnCAM(feature_conv, weight_softmax, class_idx):
     # generate the class activation maps upsample to 256x256