INNER CODE UNIT · Python

example_image

royshil/obs-backgroundremoval · Models/MediaPipeLandscape/CoreML/GenerateModel.py:57

example_image = torch.rand(*IMAGE_SHAPE)

with torch.no_grad():
    traced_model = torch.jit.trace(segmenter, example_image)

mlpackage_model = ct.convert(
    traced_model,
    source="pytorch",
    inputs=[
        ct.TensorType(
            name="pixel_values",
            shape=IMAGE_SHAPE,
            dtype=types.fp16,
        ),
    ],
    outputs=[
        ct.TensorType(
            name="alphas",

View source record →