search

hisfog / SfMNeXt-Impl

[AAAI 2024] Official implementation of "SQLdepth: Generalizable Self-Supervised Fine-Structured Monocular Depth Estimation", and more.
MIT License
91 stars 12 forks source link

FP 16 Pretrained MODEL #27

Closed Choi-YeongJoon closed 11 months ago

Choi-YeongJoon commented 11 months ago

I am very interested in your work! I ran your model in TensorRT, C++. The fps is so low that I think I need to run a model with fp16 precision. Do you have a pretrained model trained with fp16 precision? Can I possibly get it?

hisfog commented 11 months ago

The ConvNeXt model is not effcient (392.2 GMACs), you can try the KITTI (Efficient-b5) model (70.35 GMACs), which is much more faster and also accurate enough. For fp16 models, they should be trained from scratch in fp16 precision, I didn't do that training, since this repo is not for industrial use cases.