Параметры запуска#
Выбор параметров обучения осуществляется в файле config\config.yaml
. Пример файла конфигурации:
### Basic
Mode: 1 # 1 - interactive, 2 - train, 3 - eval
gpu_index: 0
### Train mode
LoadParameters: false
NumTrainRounds: -1 # -1 - infinite
NumRoundEpoch: 15
### Eval mode
NumEvalRounds: 300
### Model parameters
Model: yolo # yolo
InitWeights: true
LossRatio: 0.04
### Dataset parameters
Dataset: gen1
TimeStep: 16
NumLoadFile: 8
NumWorkers: 4
# Train dataset
BatchSize: 8
NumSteps: 32
TimeWindow: 8
# Test dataset
TestBatchSize: 64
TestNumSteps: 32
### Progress board parameters
Display: True
EveryN: 4
### Plotter parameters
PlotterThreshold: 0.8
PlotterColumns: 4
### Trainer board parameters
EpochSize: 60
Основные параметры |
|
---|---|
Название |
Описание |
|
Выбор сценария обучения (1 - interactive, 2 - train, 3 - eval) |
|
Выбор конфигурации сети (yolo) |
|
Весовой коэффициент для настройки влияния на ошибку ложных срабатываний.
Используется для расчёта ошибки следующим образом |
|
Выбор датасета (gen1, 1mpx) |
|
Временной интервал между кадрами в миллисекундах |
|
Размер пакета для обучения |
|
Количество кадров в одном примере |
|
Максимальное количество кадров, на которое может уменьшиться последовательность. Это используется для создание обучающих примеров разной длинны. |