yolo3 flir ok

This commit is contained in:
mbosi
2019-09-15 16:19:30 +02:00
parent 8c629ebe7b
commit a038e966d9
5 changed files with 21 additions and 15 deletions
+7 -7
View File
@@ -17,7 +17,7 @@ hue=.1
learning_rate=0.001
burn_in=1000
max_batches = 10000
max_batches = 20000
policy=steps
steps=8000,9000
scales=.1,.1
@@ -602,13 +602,13 @@ activation=leaky
size=1
stride=1
pad=1
filters=30
filters=24
activation=linear
[yolo]
mask = 6,7,8
anchors = 8.2087,8.5515, 18.4134,20.3391, 40.2194,29.2990, 31.6137,69.2240, 69.8497,48.3838, 108.8817,76.6316, 96.5753,145.5743, 165.9182,117.4493, 215.7497,198.4648
classes=5
classes=3
num=9
jitter=.3
ignore_thresh = .5
@@ -686,13 +686,13 @@ activation=leaky
size=1
stride=1
pad=1
filters=30
filters=24
activation=linear
[yolo]
mask = 3,4,5
anchors = 8.2087,8.5515, 18.4134,20.3391, 40.2194,29.2990, 31.6137,69.2240, 69.8497,48.3838, 108.8817,76.6316, 96.5753,145.5743, 165.9182,117.4493, 215.7497,198.4648
classes=5
classes=3
num=9
jitter=.3
ignore_thresh = .5
@@ -770,13 +770,13 @@ activation=leaky
size=1
stride=1
pad=1
filters=30
filters=24
activation=linear
[yolo]
mask = 0,1,2
anchors = 8.2087,8.5515, 18.4134,20.3391, 40.2194,29.2990, 31.6137,69.2240, 69.8497,48.3838, 108.8817,76.6316, 96.5753,145.5743, 165.9182,117.4493, 215.7497,198.4648
classes=5
classes=3
num=9
jitter=.3
ignore_thresh = .5
+1 -1
View File
@@ -11,7 +11,7 @@ int main() {
// create yolo3 model
std::string bin_path = "../tests/yolo3_flir";
int classes = 5;
int classes = 3;
tk::dnn::Yolo *yolo [3];
#include "models/Yolo3.h"