Great work! Thank you to the authors for open-sourcing it!
I found a small bug: the default path for images is incorrect. I have submitted a PR to correct the path.
Additionally, I encountered a few issues while setting up the environment that the authors might want to consider:
Running pip install onnxruntime installs the CPU version by default; it's better to specify onnxruntime-gpu.
scipy and imageio[ffmpeg] are missing and need to be installed manually.
On my Linux machine, the following versions work well: scipy==1.14.1, onnxruntime-gpu==1.18.0, and numpy==1.21.6.
Great work! Thank you to the authors for open-sourcing it!
I found a small bug: the default path for images is incorrect. I have submitted a PR to correct the path.
Additionally, I encountered a few issues while setting up the environment that the authors might want to consider:
Running
pip install onnxruntimeinstalls the CPU version by default; it's better to specifyonnxruntime-gpu.scipyandimageio[ffmpeg]are missing and need to be installed manually. On my Linux machine, the following versions work well:scipy==1.14.1, onnxruntime-gpu==1.18.0, and numpy==1.21.6.Very interesting work! :)