update

lyuwenyu · lyuwenyu · commit 8862322b3b35 · 2023-09-01T10:03:41.000Z
diff --git a/rtdetr_pytorch/README.md b/rtdetr_pytorch/README.md
@@ -4,6 +4,7 @@
 - [x] Export onnx
 - [x] Upload source code
 - [ ] Align with the [Paddle version](../rtdetr_paddle/)
+- [ ] Upload weight convert from paddle
 
 
 ## Quick start
diff --git a/rtdetr_pytorch/requirements.txt b/rtdetr_pytorch/requirements.txt
@@ -1,7 +1,7 @@
-torch>=2.0.1
-torchvision>=0.15.2
-onnx>=1.14.0
-onnxruntime>= 1.15.1
+torch==2.0.1
+torchvision==0.15.2
+onnx==1.14.0
+onnxruntime==1.15.1
 pycocotools
 PyYAML
-scipy
+scipy
