diff --git a/README.md b/README.md index 8150e7dffba69fae7c68bde6df3b37eb2efce326..e69f20711af7837b59c555a8d7857eb69eb21e9d 100644 --- a/README.md +++ b/README.md @@ -18,17 +18,8 @@ ## 相关仓库 | 目标检测模型 | 路径 | | :----- | :----- | -YoloV3 | https://github.com/bubbliiiing/yolo3-pytorch -Efficientnet-Yolo3 | https://github.com/bubbliiiing/efficientnet-yolo3-pytorch -YoloV4 | https://github.com/bubbliiiing/yolov4-pytorch -YoloV4-tiny | https://github.com/bubbliiiing/yolov4-tiny-pytorch -Mobilenet-Yolov4 | https://github.com/bubbliiiing/mobilenet-yolov4-pytorch -YoloV5-V5.0 | https://github.com/bubbliiiing/yolov5-pytorch -YoloV5-V6.1 | https://github.com/bubbliiiing/yolov5-v6.1-pytorch -YoloX | https://github.com/bubbliiiing/yolox-pytorch -YoloV7 | https://github.com/bubbliiiing/yolov7-pytorch -YoloV7-tiny | https://github.com/bubbliiiing/yolov7-tiny-pytorch YoloV7-OBB | https://github.com/Egrt/yolov7-obb +YoloV7-Tiny-OBB | https://github.com/Egrt/yolov7-tiny-obb ## 性能情况 | 训练数据集 | 权值文件名称 | 测试数据集 | 输入图片大小 | mAP 0.5 | diff --git a/requirements.txt b/requirements.txt index b3197cd348605c544ef2a7145db8565b5378b4f5..c528fd56eaa58445b5c1b3dc142ba5b175f6c3e7 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,8 +2,8 @@ scipy==1.2.1 numpy==1.17.0 matplotlib==3.1.2 opencv_python==4.1.2.30 -torch==1.12.1 -torchvision==0.13.1 +torch==1.10.1 +torchvision==0.11.2 tqdm==4.60.0 Pillow==8.2.0 h5py==2.10.0