Skip to content

[MXNet] detection and semantic segmentation

Valentina Kustikova requested to merge github/fork/ismukhin/mxnet into master

Created by: ismukhin

@valentina-kustikova, добрый день! На данном этапе получилось сделать поддержку всех моделей детектирования у MXNet и одно семейство моделей сегментации. Но с детектированием небольшая проблема: для отрисовки боксов в io.adapter происходит умножение на значение ширины и высоты, а выход моделей MXNet этого не требует. Т.е. у меня два варианта решения этой проблемы:

  1. Чтобы не было повторов кода, после инференса перед обработкой выхода намеренно делить на значение ширины и высоты, так получится спокойно использовать DetectionIO
  2. Ввести в обиход task'у detection_mxnet и сделать копипаст обработки выхода без умножения на ширину и высоту

А вот с остальными задачи пока что проблема, поскольку они падают или из-за непонятных input_shapes, например так:

 MXNetError: Check failed: shape_assign(&(*in_shape)[i], dshape): Incompatible input shape: expected [1,-1,4,4], got [1,256,60,60]

Или же из-за странных ошибок, например: MXNetError: Check failed: *end <= axis_size: Invalid end for end=1000 as axis_size is 240

Merge request reports