Я делаю некоторую семантическую сегментацию людей, используя млмодель DeepLabV3. Результат после прогнозирования — 513*513 в виде MLMultiArray. В настоящее время я изменяю размер вывода камеры до этого размера, чтобы применить сегментированный массив.
Как я могу изменить размер MLMultiArray, чтобы он соответствовал размеру текстуры моей камеры?
if let observations = request.results as? [VNCoreMLFeatureValueObservation],
let segmentationmap = observations.first?.featureValue.multiArrayValue {
// row - 513, col - 513
guard let row = segmentationmap.shape[0] as? Int,
let col = segmentationmap.shape[1] as? Int else {
return
}
}
Вот некоторый демонстрационный код для использования DeepLab V3 в приложении для iOS: https://github.com/hollance/SemanticSegmentationMetalDemo
Это именно то, что мне было нужно, я закончил до сегментации и боролся с высотой текстуры, большое спасибо.
Если вы можете порекомендовать мне такие учебные материалы, я тоже могу их использовать.