Skip to content

tried to fix yolo models

Evgeni Vasilev requested to merge github/fork/VankaVstanka03/yolo_models_adapter into master

Created by: VankaVstanka03

@FenixFly , добавил фикс, работают следующие модели:

  • yolo-v1-tiny-tf
  • yolo-v2-tf
  • yolo-v2-tiny-tf
  • yolo-v3-tf
  • yolo-v3-tiny-tf

Хотел попытаться приспособить yolo 4 версий, но пока выдает ошибку обращения к NoneType элементу. Модели с суффиксами onnx, tiny-onnx довольно сильно отличаются выходными данными, поэтому, скорее всего, придётся много чего переопределить

Merge request reports