質問編集履歴
3
具体的なコードの追加
test
CHANGED
|
File without changes
|
test
CHANGED
|
@@ -263,3 +263,57 @@
|
|
|
263
263
|
# 追記 roi_headのコード
|
|
264
264
|
|
|
265
265
|
https://github.com/pytorch/vision/blob/10d5a55c332771164c13375f445331c52f8de6f1/torchvision/models/detection/roi_heads.py
|
|
266
|
+
|
|
267
|
+
|
|
268
|
+
|
|
269
|
+
#追記 FC7の部分のコード
|
|
270
|
+
|
|
271
|
+
```
|
|
272
|
+
|
|
273
|
+
class TwoMLPHead(nn.Module):
|
|
274
|
+
|
|
275
|
+
"""
|
|
276
|
+
|
|
277
|
+
Standard heads for FPN-based models
|
|
278
|
+
|
|
279
|
+
Arguments:
|
|
280
|
+
|
|
281
|
+
in_channels (int): number of input channels
|
|
282
|
+
|
|
283
|
+
representation_size (int): size of the intermediate representation
|
|
284
|
+
|
|
285
|
+
"""
|
|
286
|
+
|
|
287
|
+
|
|
288
|
+
|
|
289
|
+
def __init__(self, in_channels, representation_size):
|
|
290
|
+
|
|
291
|
+
super(TwoMLPHead, self).__init__()
|
|
292
|
+
|
|
293
|
+
|
|
294
|
+
|
|
295
|
+
self.fc6 = nn.Linear(in_channels, representation_size)
|
|
296
|
+
|
|
297
|
+
self.fc7 = nn.Linear(representation_size, representation_size)
|
|
298
|
+
|
|
299
|
+
|
|
300
|
+
|
|
301
|
+
def forward(self, x):
|
|
302
|
+
|
|
303
|
+
x = x.flatten(start_dim=1)
|
|
304
|
+
|
|
305
|
+
|
|
306
|
+
|
|
307
|
+
x = F.relu(self.fc6(x))
|
|
308
|
+
|
|
309
|
+
x = F.relu(self.fc7(x))
|
|
310
|
+
|
|
311
|
+
|
|
312
|
+
|
|
313
|
+
return x
|
|
314
|
+
|
|
315
|
+
```
|
|
316
|
+
|
|
317
|
+
コードの全体は以下です。
|
|
318
|
+
|
|
319
|
+
https://github.com/pytorch/vision/blob/10d5a55c332771164c13375f445331c52f8de6f1/torchvision/models/detection/faster_rcnn.py
|
2
roi_headのコード
test
CHANGED
|
File without changes
|
test
CHANGED
|
@@ -24,6 +24,8 @@
|
|
|
24
24
|
|
|
25
25
|
torchvisionのモジュールとなっているためあまり自分の手で変更はしたくありません、、何か方法はありますでしょうか、、
|
|
26
26
|
|
|
27
|
+
(もしなければroi_headのコードを一番下に添付しますので具体的な方法を教えて下さると助かります。)
|
|
28
|
+
|
|
27
29
|
|
|
28
30
|
|
|
29
31
|
### ソースコード（model定義部分）
|
|
@@ -255,3 +257,9 @@
|
|
|
255
257
|
ご教示いただければ非常に助かります。
|
|
256
258
|
|
|
257
259
|
よろしくおねがいします。
|
|
260
|
+
|
|
261
|
+
|
|
262
|
+
|
|
263
|
+
# 追記 roi_headのコード
|
|
264
|
+
|
|
265
|
+
https://github.com/pytorch/vision/blob/10d5a55c332771164c13375f445331c52f8de6f1/torchvision/models/detection/roi_heads.py
|
1
フックの方法、、
test
CHANGED
|
File without changes
|
test
CHANGED
|
@@ -18,6 +18,14 @@
|
|
|
18
18
|
|
|
19
19
|
|
|
20
20
|
|
|
21
|
+
追記:
|
|
22
|
+
|
|
23
|
+
どうやらフックで引っ掛ける方法があるようですが、このroi_headなどは
|
|
24
|
+
|
|
25
|
+
torchvisionのモジュールとなっているためあまり自分の手で変更はしたくありません、、何か方法はありますでしょうか、、
|
|
26
|
+
|
|
27
|
+
|
|
28
|
+
|
|
21
29
|
### ソースコード（model定義部分）
|
|
22
30
|
|
|
23
31
|
|
|
@@ -242,7 +250,7 @@
|
|
|
242
250
|
|
|
243
251
|
|
|
244
252
|
|
|
245
|
-
#####modelがここまできれいに出力できるならこのroiheadsの出力を取り出すことはできないでしょうか？？
|
|
253
|
+
#####modelがここまできれいに出力できるならこのroi_headsの出力(fc7まで)を取り出すことはできないでしょうか？？
|
|
246
254
|
|
|
247
255
|
ご教示いただければ非常に助かります。
|
|
248
256
|
|