iOS : 2D Float 데이터를 수동으로 MTLTexture 또는 MPSImage로 설정하는 방법은 무엇입니까?

Question

내가 수동으로 값을 MPSImage 또는 2D 금속 질감을 작성하고 길쌈 네트워크 작업을 할 것을 전달하는 데 노력하고있어. CNN (금속 성능 쉐이더)에 대한 입력 내가 CGContext의 UnsafePointer을 전달할 수 그 이유는, 일반적으로 (이 https://developer.apple.com/library/content/samplecode/MPSCNNHelloWorld/Introduction/Intro.html#//apple_ref/doc/uid/TP40017482-Intro-DontLinkElementID_2 같은) 이미지이지만, 이번에는 내가 입력으로 float 배열을 사용하고 싶습니다.

다음은 내가 시도한 것입니다. 입력 배열을 NSData으로 변환했지만 작동하지 않았습니다.

var inputData = NSData(bytes: inputFloatArrayOfArray, length: inputFloatArrayOfArray.count * inputFloatArrayOfArray[0].count * MemoryLayout<Float>.size) 
// The type of inputFloatArrayOfArray is [[Float]] 
network.srcImage.texture.replace(region: MTLRegion(origin: MTLOrigin(x: 0, y: 0, z: 0), 
                size: MTLSize(width: inputWidth, height: inputHeight, depth: 1)), 
           mipmapLevel: 0, 
           slice: 0, 
           withBytes: &inputData, 
           bytesPerRow: inputWidth * MemoryLayout<Float>.size, 
           bytesPerImage: 0) 

Manually set a 1D Texture in Metal

는 내 질문에 관련 될 수있다 (참고 :이 "생의 바이트 데이터에로드 된 UIImage 변환하지만, 더미 UIImage을 만들어 텍스처를로드 2D 텍스처와 거래 나를 위해 해킹 같은 느낌."라고),하지만 충분한 대답이없는 것 같습니다. 이제 나는이 문제를 해결할 아이디어가 없습니다. 아이디어가 있으면 알려 주시기 바랍니다.

대단히 감사합니다. 당신의 텐서가 < = 4 기능 채널이있는 경우

Answer 1

는, 당신은 RGBA는 텍스처에있을 것입니다 경우 기능 채널 0-3 앉아 그들에 복사합니다. 만약 당신의 텐서가 그 이상이라면 MTLTexture2DArray를 사용하십시오. 처음 4 개 이후의 추가 기능 채널은 배열의 나중 이미지에서 연속적으로 동일한 좌표로 이동합니다.

Image[0]: // a.k.a. slice 0 
    pix[0][0] = {feature channel 0, feature channel 1, feature channel 2, feature channel 3} 

Image[1]: // a.k.a. slice 1 
    pix[0][0] = {feature channel 4, feature channel 5, feature channel 6, feature channel 7} 

Image[2]: // a.k.a. slice 2 
    pix[0][0] = {feature channel 8, feature channel 9, feature channel 10, feature channel 11}