Troubleshooting Issues with Colorization Model Conversion: Seeking Suggestions

Hello everyone,

I encountered some compiler errors while following a WWDC video on converting a colorization PyTorch model to CoreML. I have followed all the steps correctly, but I'm facing issues with the following lines of code provided in the video:

1. In the colorize() method, there is a line:

let modelInput = try ColorizerInput(inputWith: lightness.cgImage!)

This line expects a cgImage as input, but the auto-generated Model class only accepts an MLMultiArray or MLShapedArray, not an image. Video conversion step did not cover setting the input or output as ImageType.

1. In the extractColorChannels() method, there are a couple of lines:

let outA: [Float] = output.output_aShapedArray.scalars
let outB: [Float] = output.output_bShapedArray.scalars

However, I only have output.var183_aShapedArray available. In other words, there is no var183_bShapedArray.

I would appreciate any thoughts or suggestions you may have regarding these issues. Thank you.

Link to the WWDC22 session 10017 https://developer.apple.com/videos/play/wwdc2022/10017/