Open3D-ML — сегментация облаков точек и 3D-детекция

Open3D-ML — расширение библиотеки Open3D для задач машинного обучения с 3D-данными. Оно предоставляет инструменты для обработки облаков точек и других 3D-структур, ориентированные на исследователей и разработчиков, работающих с 3D-моделированием и компьютерным зрением.

Возможности и состав проекта

Проект поддерживает сегментацию облаков точек и 3D-обнаружение объектов с использованием предобученных моделей. Open3D-ML интегрируется с фреймворками TensorFlow и PyTorch, предлагая пайплайны для обучения моделей и средства для визуализации 3D-данных. Это обеспечивает гибкость при работе с существующими проектами.

Начало работы и требования

Open3D-ML входит в Open3D; README предлагает установить open3d и выбрать файл requirements для совместимого PyTorch или TensorFlow. Для GPU есть отдельные варианты зависимостей. После установки проверьте импорт соответствующего open3d.ml backend; если версии фреймворка или ускорителя отличаются от поддерживаемых сборок, может понадобиться сборка Open3D из исходников.

Ограничения и выбор сценария

Версии Open3D, ML-фреймворка и GPU-runtime должны быть согласованы; номер CUDA из старого примера не является универсальным требованием. Предобученные модели привязаны к задачам и наборам данных. Перед обработкой своих сканов проверьте масштаб, признаки точек и классы, иначе формально успешный инференс может дать неподходящую разметку.

Лицензия и условия использования

В проверенном LICENSE Open3D-ML указан MIT, несмотря на неопределённую классификацию GitHub. Перед распространением сохраните уведомления об авторстве и лицензию; условия датасетов, checkpoints и отдельных зависимостей проверяются дополнительно.