still onnx file export issue

[KaiyangQ99]

I want to have a nice onnx file to know its MACs, params, and probably run it in some NPUs.
I struggled really hard but still bugs here.

[KaiyangQ99]

or at least please tell me for NYUs dataset: its input resolution, MACs, and params number.

[rhmiao]

Usually, the data generated after running process in nyu_dataset will contain img data
https://github.com/megvii-research/OccDepth/blob/main/occdepth/data/NYU/nyu_dataset.py#L251