質問編集履歴
2
説明が不十分だったため
test
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
chainercv, faster R-CNNで
|
|
1
|
+
chainercv, faster R-CNNでクラス数を変更し，再度学習したい
|
test
CHANGED
|
@@ -2,11 +2,11 @@
|
|
|
2
2
|
|
|
3
3
|
|
|
4
4
|
|
|
5
|
-
|
|
5
|
+
タイトル，質問の内容ともに説明が不十分だったので修正しました．
|
|
6
|
-
|
|
7
|
-
（クラス数は変化する可能性がある．）
|
|
8
6
|
|
|
9
7
|
|
|
8
|
+
|
|
9
|
+
一度学習した物体検出モデルのクラス数変更のため，学習済みモデルを元に再度学習を行いたい．
|
|
10
10
|
|
|
11
11
|
pretrained_modelに'imagenet'を設定した場合には最終層の重みはコピーされないのに対し，
|
|
12
12
|
|
|
@@ -56,7 +56,9 @@
|
|
|
56
56
|
|
|
57
57
|
File "<__array_function__ internals>", line 6, in copyto
|
|
58
58
|
|
|
59
|
-
ValueError: could not broadcast input array from shape (101
|
|
59
|
+
ValueError: could not broadcast input array from shape (101) into shape (102)
|
|
60
|
+
|
|
61
|
+
|
|
60
62
|
|
|
61
63
|
|
|
62
64
|
|
|
@@ -68,7 +70,7 @@
|
|
|
68
70
|
|
|
69
71
|
|
|
70
72
|
|
|
71
|
-
```pytho
|
|
73
|
+
```python
|
|
72
74
|
|
|
73
75
|
import chainer
|
|
74
76
|
|
|
@@ -92,7 +94,23 @@
|
|
|
92
94
|
|
|
93
95
|
chainer.serializers.save_npz('tmp.npz', train_chain.faster_rcnn)
|
|
94
96
|
|
|
97
|
+
|
|
98
|
+
|
|
99
|
+
# 検出したい物体が増えたので，上記のモデルを元に再度学習したい
|
|
100
|
+
|
|
95
|
-
model = chainercv.links.FasterRCNNVGG16(n_fg_class=10, pretrained_model='tmp.npz')
|
|
101
|
+
model = chainercv.links.FasterRCNNVGG16(n_fg_class=101, pretrained_model='tmp.npz')
|
|
102
|
+
|
|
103
|
+
train_chain = chainercv.links.model.faster_rcnn.FasterRCNNTrainChain(model)
|
|
104
|
+
|
|
105
|
+
|
|
106
|
+
|
|
107
|
+
''''
|
|
108
|
+
|
|
109
|
+
再度trainする
|
|
110
|
+
|
|
111
|
+
'''
|
|
112
|
+
|
|
113
|
+
|
|
96
114
|
|
|
97
115
|
|
|
98
116
|
|
1
誤投稿のため編集
test
CHANGED
|
File without changes
|
test
CHANGED
|
@@ -10,7 +10,9 @@
|
|
|
10
10
|
|
|
11
11
|
pretrained_modelに'imagenet'を設定した場合には最終層の重みはコピーされないのに対し，
|
|
12
12
|
|
|
13
|
+
セーブしたnpzファイルをロードするとすべての重みをコピーしようとするため，エラーが生じるのだと思います．
|
|
14
|
+
|
|
13
|
-
|
|
15
|
+
コピーできる重みだけをコピーするにはどうすればよいでしょうか．
|
|
14
16
|
|
|
15
17
|
|
|
16
18
|
|
|
@@ -100,20 +102,14 @@
|
|
|
100
102
|
|
|
101
103
|
|
|
102
104
|
|
|
103
|
-
|
|
104
|
-
|
|
105
|
-
|
|
106
|
-
|
|
107
|
-
### 試したこと
|
|
108
|
-
|
|
109
|
-
|
|
110
|
-
|
|
111
|
-
ここに問題に対して試したことを記載してください。
|
|
112
|
-
|
|
113
|
-
|
|
114
|
-
|
|
115
105
|
### 補足情報（FW/ツールのバージョンなど）
|
|
116
106
|
|
|
117
107
|
|
|
118
108
|
|
|
109
|
+
chainer.__version__
|
|
110
|
+
|
|
111
|
+
'7.2.0'
|
|
112
|
+
|
|
119
|
-
|
|
113
|
+
chainercv.__version__
|
|
114
|
+
|
|
115
|
+
'0.13.1'
|