Update README.md

ece1cca1 · JiaQi Xu · GitHub · 02b1c964 · ece1cca1
隐藏空白更改
内联并排

Showing with 1 addition and 2 deletion

README.md README.md +1 -2

未找到文件。
--- a/README.md
+++ b/README.md
@@ -17,8 +17,7 @@
 - [ ] ……balabla

 ### 所需环境
-tensorflow-gpu==1.13.1  
-keras==2.1.5  
+torch=1.2.0

 ### 注意事项
 代码中的yolo4_weights.h5是基于608x608的图片训练的，但是由于显存原因。我将代码中的图片大小修改成了416x416。有需要的可以修改回来。 代码中的默认anchors是基于608x608的图片的。