From c623d56e1f859caa3002d678dad26ffddf2b26d8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=99=BD=E9=B9=AD=E5=85=88=E7=94=9F?= <766529835@qq.com> Date: Mon, 20 Feb 2023 22:04:59 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=E8=AF=B4=E6=98=8E?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index e69f207..19ce1d3 100644 --- a/README.md +++ b/README.md @@ -28,7 +28,8 @@ YoloV7-Tiny-OBB | https://github.com/Egrt/yolov7-tiny-obb ### 预测结果展示 ![预测结果](img/test.jpg) ## 所需环境 -torch==1.2.0 +torch==1.10.1 +torchvision==0.11.2 为了使用amp混合精度,推荐使用torch1.7.1以上的版本。 ## 文件下载 -- GitLab