react-native-executorch/lib/hooks/computer_vision/useImageSegmentation.js

An easy way to run AI models in React Native with ExecuTorch

import { useNonStaticModule } from '../useNonStaticModule';
import { ImageSegmentationModule } from '../../modules/computer_vision/ImageSegmentationModule';
export const useImageSegmentation = ({ model, preventLoad = false }) => useNonStaticModule({
    module: ImageSegmentationModule,
    model,
    preventLoad,
});