요약

하루종일 Input dimension 바꿀 방법 찾는데 못하고있음
Pytorch to ONNX : ONNX로 변환할 때 방법이 일단 Pytorch 모델을 한 번 Inference하고 그 과정을 Graph로 기록하는 것이기에, Pytorch 모델에 맞는 Input shape을 넣어줘야 가능함.
ONNX to Tensorflow : 변환 함수에 Axis 변경 기능이 없음
Tensorflow to TFLite : 여기도 변환하는 과정에 Axis 변경 기능 없는듯
TFLite 파일에 직접 접근해 layer 삭제 -> 애초에 불가능한듯?
TFLite 파일에 대한 Input인 TensorImage 클래스를 변경 -> 일단 기존 TensorImage 인스턴스를 변경시키는 걸 권장하지 않는다고 공식문서에 나와있음. 건드리기도 쉽지않음.

현재 가능해보이는 방법 : 애초에 Pytorch 모델을 Tensorflow input shape으로 image를 넣도록 custom하면 가능할까? 다만 이 방법도 정상적으로 .tflite 파일에서 transpose없이 될지는 해봐야 앎.