mirrors
/
yolov7
mirror of https://github.com/WongKinYiu/yolov7.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

feat(Dockerfile): add preweights for training

pull/2127/head
barisx 2025-04-14 18:41:57 +03:00
parent f6ddbc6ee2
commit 491711c114
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
Dockerfile
View File

@ -28,4 +28,10 @@ RUN rm -rf ./workspace
RUN bash scripts/build_opencv.sh
# Test CUDA and OpenCV support
RUN bash scripts/test-cmds.sh
RUN bash scripts/test-cmds.sh
RUN wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7.pt # Recommend for lower specs
# RUN wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7x.pt
# RUN wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-w6.pt
# RUN wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-e6.pt
# RUN wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-d6.pt
RUN wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-e6e.pt # Recommend for high specs