質問編集履歴
2
タイトルの修正
test
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
PyTorchを使ったネットワークの解釈を手伝ってほしいです
|
|
1
|
+
PyTorchを使ったネットワーク（20行程度）の解釈を手伝ってほしいです
|
test
CHANGED
|
File without changes
|
1
コードの無関係な部分を削除しました．
test
CHANGED
|
File without changes
|
test
CHANGED
|
@@ -102,32 +102,6 @@
|
|
|
102
102
|
|
|
103
103
|
return self.fc(x)
|
|
104
104
|
|
|
105
|
-
|
|
106
|
-
|
|
107
|
-
def save_attention_mask(self, x, path):
|
|
108
|
-
|
|
109
|
-
B = x.shape[0]
|
|
110
|
-
|
|
111
|
-
self.forward(x)
|
|
112
|
-
|
|
113
|
-
x = x.cpu() * torch.Tensor([0.229, 0.224, 0.225]).reshape(-1, 1, 1)
|
|
114
|
-
|
|
115
|
-
x = x + torch.Tensor([0.485, 0.456, 0.406]).reshape(-1, 1, 1)
|
|
116
|
-
|
|
117
|
-
fig, axs = plt.subplots(4, 2, figsize=(6, 8))
|
|
118
|
-
|
|
119
|
-
plt.axis('off')
|
|
120
|
-
|
|
121
|
-
for i in range(4):
|
|
122
|
-
|
|
123
|
-
axs[i, 0].imshow(x[i].permute(1, 2, 0))
|
|
124
|
-
|
|
125
|
-
axs[i, 1].imshow(self.mask_[i][0])
|
|
126
|
-
|
|
127
|
-
plt.savefig(path)
|
|
128
|
-
|
|
129
|
-
plt.close()
|
|
130
|
-
|
|
131
105
|
```
|
|
132
106
|
|
|
133
107
|
|