回答編集履歴
1
補足を追加
test
CHANGED
|
@@ -1 +1,17 @@
|
|
|
1
1
|
ええと、create_custom_model.shの引数で指定するのは「クラス数」(number of classes→num_classes)の数値(整数)です。20クラスあるなら「20」だけを指定するってことです。
|
|
2
|
+
|
|
3
|
+
|
|
4
|
+
|
|
5
|
+
[PyTorch-YOLOv3/create_custom_model.sh at master · eriklindernoren/PyTorch-YOLOv3](https://github.com/eriklindernoren/PyTorch-YOLOv3/blob/master/config/create_custom_model.sh)
|
|
6
|
+
|
|
7
|
+
|
|
8
|
+
|
|
9
|
+
```bash
|
|
10
|
+
|
|
11
|
+
NUM_CLASSES=$1
|
|
12
|
+
|
|
13
|
+
(中略)
|
|
14
|
+
|
|
15
|
+
filters=$(expr 3 * $(expr $NUM_CLASSES + 5))
|
|
16
|
+
|
|
17
|
+
```
|