diff --git a/.DS_Store b/.DS_Store new file mode 100644 index 0000000..3cd0c1a Binary files /dev/null and b/.DS_Store differ diff --git a/class_activation_map.py b/class_activation_map.py index e0a8f33..eeb7507 100644 --- a/class_activation_map.py +++ b/class_activation_map.py @@ -65,12 +65,6 @@ def generate_CAM(self, img, class_code = None): if class_code is None: class_code = np.argmax(output.cpu().data.numpy()) - # one_hot = np.zeros((1, output.size()[-1]), dtype = np.float32) - # one_hot[0][class_code] = 1 - # one_hot = torch.from_numpy(one_hot) - # one_hot.requires_grad = True - # one_hot = one_hot.cuda() - # activation_value = torch.sum(torch.mm(one_hot, output.t())) activation_value = output[0][class_code] # reset grad diff --git a/presentation/conf_mat_recall.png b/presentation/conf_mat_recall.png new file mode 100644 index 0000000..8fed288 Binary files /dev/null and b/presentation/conf_mat_recall.png differ diff --git a/presentation/conf_max.png b/presentation/conf_max.png new file mode 100644 index 0000000..58c8c4f Binary files /dev/null and b/presentation/conf_max.png differ diff --git a/presentation/conf_max_precision.png b/presentation/conf_max_precision.png new file mode 100644 index 0000000..ece6438 Binary files /dev/null and b/presentation/conf_max_precision.png differ