Skip to content

Commit 0373dff

Browse files
committed
Update Dockerfile for Yolo workflow
Signed-off-by: Beat Buesser <[email protected]>
1 parent 493ad6f commit 0373dff

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

.github/actions/yolo/Dockerfile

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,4 +24,6 @@ RUN pip install pytorchyolo==1.8.0 tensorflow==2.14.1 scikit-learn==1.4.2 pytest
2424

2525
RUN cd /tmp/ && git clone https://github.com/eriklindernoren/PyTorch-YOLOv3.git && cd ./PyTorch-YOLOv3/weights && ./download_weights.sh
2626

27+
RUN mkdir /tmp/yolo_v8.3.0 && cd /tmp/yolo_v8.3.0 && wget https://github.com/ultralytics/assets/releases/download/v8.3.0/yolov8n.pt && wget https://github.com/ultralytics/assets/releases/download/v8.3.0/yolov10n.pt
28+
2729
CMD ["/bin/bash"]

0 commit comments

Comments
 (0)