Ошибка вывода openvino yolo-v3 не может преобразовать число с плавающей запятой в целое число - PullRequest
3 голосов
/ 13 апреля 2020

Здравствуйте, я выполнил все шаги, чтобы сделать вывод и успешно запустить его на модели по этой ссылке: https://pjreddie.com/media/files/yolov3.weights, но когда я попробовал это на модели, я тренировался с Дар knet i получите эту ошибку:

[ INFO ] Creating Inference Engine...
[ INFO ] Loading network files:
    newyolo.xml
    newyolo.bin
[ INFO ] Preparing inputs
[ INFO ] Loading model to the plugin
[ INFO ] Starting inference...
To close the application, press 'CTRL+C' here or switch to the output window and press ESC key
To switch between sync/async modes, press TAB key in the output window
yolo_original.py:280: DeprecationWarning: shape property of IENetLayer is deprecated. Please use shape property of DataPtr instead objects returned by in_data or out_data property to access shape of input or output data on corresponding ports
  out_blob = out_blob.reshape(net.layers[net.layers[layer_name].parents[0]].shape)
[ INFO ] Layer detector/yolo-v3/Conv_14/BiasAdd/YoloRegion parameters: 
[ INFO ]          classes : 10
[ INFO ]          num     : 3
[ INFO ]          coords  : 4
[ INFO ]          anchors : [55.0, 56.0, 42.0, 87.0, 68.0, 81.0]
Traceback (most recent call last):
  File "yolo_original.py", line 363, in <module>
    sys.exit(main() or 0)
  File "yolo_original.py", line 286, in main
    args.prob_threshold)
  File "yolo_original.py", line 153, in parse_yolo_region
    h_scale=orig_im_h, w_scale=orig_im_w))
  File "yolo_original.py", line 99, in scale_bbox
    xmin = int((x - w / 2) * w_scale)
ValueError: cannot convert float NaN to integer

зная, что я предоставил правильную форму и изменил yolo_v3. json, чтобы соответствовать моей модели, вот содержимое моего yolo_v3. json:

[
  {
    "id": "TFYOLOV3",
    "match_kind": "general",
    "custom_attributes": {
      "classes": 10,
      "anchors": [18,22,31,33,33,50,55, 56,42,87,68,81,111,98,73,158,156,202],
      "coords": 4,
      "num": 9,
      "masks":[[6, 7, 8], [3, 4, 5], [0, 1, 2]],
      "entry_points": ["detector/yolo-v3/Reshape", "detector/yolo-v3/Reshape_4", "detector/yolo-v3/Reshape_8"]
    }
  }
]

Я пробовал несколько вещей для отладки, например, не предоставляя jsonfile .... et c

ps: yolo_original.py - это то же самое демо, которое поставляется с openvino, только что переименованным, я использую В openvino версии 2020.1 преобразование NaN в плавающее или пропуск значений с помощью Nan не решило проблему.

...