Преобразование сгенерированных контрольных точек в .weights darkflow - PullRequest
Новая
       6

Преобразование сгенерированных контрольных точек в .weights darkflow

0 голосов
/ 31 мая 2018

Я провел индивидуальное обучение с использованием Darkflow и yolov2.weights.В папке контрольных точек есть четыре файла. Это: 

1.yolov2-3c-5500.data-00000-of-00001
2.yolov2-3c-5500.index
3.yolov2-3c-5500.meta
4.yolov2-3c-5500.profile

Я хочу преобразовать эти файлы .weights?

В настоящее время контрольные точки загружаются так: 

options = {
    'model': 'cfg/yolov2-4c.cfg',
    'load': 5500,
    'threshold': 0.3,
    'gpu':1.0

}

Я хочу загрузить ключ загрузки в файл .weights.

1 Ответ

0 голосов
/ 08 июня 2018

Обученные файлы контрольных точек yolov2-3c-5500.data Файл фактически является вашим файлом веса.Если вы хотите преобразовать в файл .pb, используйте следующую команду 

flow --model cfg/Your_ConfigFile.cfg --load ckpt/Your_required_checkpoint-chkptNumber --savepb

Пример: flow --model cfg / Your_ConfigFile.cfg --load 5500 --savepb

...