• Регистрация
fatbad
fatbad 0.00
н/д

yolo2 не тренеруется

24.05.2020
Коллеги, добрый день.   после или во время выполения [detector,info] = trainYOLOv2ObjectDetector(ds,lgraph,options); выдает следующую ошибку Error using trainYOLOv2ObjectDetector>iParseInputsY...

Коллеги, добрый день.

 

после или во время выполения

[detector,info] = trainYOLOv2ObjectDetector(ds,lgraph,options);

выдает следующую ошибку

Error using trainYOLOv2ObjectDetector>iParseInputsYolov2 (line 244)
Invalid network.

Error in trainYOLOv2ObjectDetector (line 174)
[trainingData, lgraph, params, options] = iParseInputsYolov2(...

Caused by:
    Network: The input to the YOLO v2 transform layer must have 60 channels to support 4 anchor boxes and 10 classes. The number of channels must equal numAnchors * (5 + numClasses). Update the training data, the number of anchor boxes specified in the yolov2Transform layer, or the layers preceding the transform layer.

Подсдкажите, пожалуйста, подкажите, что значит "transform layer must have 60 channels to support 4 anchor boxes and 10 classes"?

И саммое главное, как это сделать?

 

С одним классом работает. Все как в примере.

Теги

    24.05.2020

    Комментарии

    • fatbad
      fatbad0.00
      24.05.2020 17:28

      Сетку загружал из примера. Вот так пробовал лечить

      lgraph = replaceLayer(lgraph, 'yolov2ClassConv',convolution2dLayer(1,60));
      lgraph.Layers(23)

      Ни чего не получается.