fix: add requirements.txt

2024-12-22 12:22:38 +00:00 · 2024-07-04 05:10:42 +08:00 · 2024-07-04 05:10:42 +08:00 · 05511c2f02
commit 05511c2f02
parent 3084170142
1 changed files with 22 additions and 0 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -0,0 +1,22 @@
+--extra-index-url https://download.pytorch.org/whl/cu118 
+torch==2.3.0 
+torchvision==0.18.0 
+torchaudio==2.3.0 
+
+numpy==1.26.4
+pyyaml==6.0.1
+opencv-python==4.10.0.84
+scipy==1.13.1
+imageio==2.34.2
+lmdb==1.4.1
+tqdm==4.66.4
+timm==1.0.7
+rich==13.7.1
+ffmpeg==1.4
+onnxruntime-gpu==1.18.0
+onnx==1.16.1
+scikit-image==0.24.0
+albumentations==1.4.10
+matplotlib==3.9.0
+imageio-ffmpeg==0.5.1
+tyro==0.8.5