Topst ai npu 사용 문의

AI
1

안녕하세요.

npu 0번의 출력을 1번이 받아서 추론이 가능할까요?
예를 들어, 네트워크의 인코더를 0번에 올리고 디코더를 1번에 올린다음 이미지를 0번에 넣고 추론하고, 출력 텐서를 1번에 다시 넣어서 추론이 가능한지 궁금합니다.

더 정확히는, 이미지 → 0번 npu → 0번 출력 텐서(translaion/rotation matrix)
이미지x(0번 출력 텐서) → 1번 npu → 출력
0번 npu에서 이미지를 입력받아 이미지에 shifting을 가할 이동/회전 행렬을 추론하고, 추론된 행렬로 이미지를 shifting하고 shifting된 이미지로 1번 npu에서 추론하고 싶습니다.

감사합니다.

2

안녕하세요.
출력 텐서를 받아서 1번 npu의 입력 버퍼에 그대로 넣으시면 가능하실 것 같습니다.
정확히 어떤 모델을 사용하시려는 걸까요?

3

답변 감사합니다. 사용할 모델은 ufldv2로 이미 compile에 성공하여서 문제는 없을 것으로 예상됩니다. 다만, 출력 이미지에 전처리를 수행할 때 함수를 preprocess.c에 생성 후 사용하면 될까요?

감사합니다.

4

네, postproc.c에서 수정 진행하시면 됩니다.
감사합니다.