Commit
•
17e4926
1
Parent(s):
11b462e
updates
Browse files- Dockerfile +2 -1
- models/yolov5x.yaml +1 -1
Dockerfile
CHANGED
@@ -40,7 +40,8 @@ COPY . /usr/src/app
|
|
40 |
# sudo docker run --gpus all --ipc=host ultralytics/yolov5:latest while true; do python3 train.py --evolve; done
|
41 |
|
42 |
# Bash into running container
|
43 |
-
# sudo docker container exec -it
|
|
|
44 |
|
45 |
# Bash into stopped container
|
46 |
# sudo docker commit 6d525e299258 user/test_image && sudo docker run -it --gpus all --ipc=host -v "$(pwd)"/coco:/usr/src/coco --entrypoint=sh user/test_image
|
|
|
40 |
# sudo docker run --gpus all --ipc=host ultralytics/yolov5:latest while true; do python3 train.py --evolve; done
|
41 |
|
42 |
# Bash into running container
|
43 |
+
# sudo docker container exec -it ba65811811ab bash
|
44 |
+
# python -c "from utils.utils import *; create_backbone('weights/last.pt')" && gsutil cp weights/backbone.pt gs://*
|
45 |
|
46 |
# Bash into stopped container
|
47 |
# sudo docker commit 6d525e299258 user/test_image && sudo docker run -it --gpus all --ipc=host -v "$(pwd)"/coco:/usr/src/coco --entrypoint=sh user/test_image
|
models/yolov5x.yaml
CHANGED
@@ -26,7 +26,7 @@ backbone:
|
|
26 |
|
27 |
# yolov5 head
|
28 |
head:
|
29 |
-
[[-1, 3, Bottleneck, [1024]], # 11
|
30 |
[-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1, 0]], # 12 (P5/32-large)
|
31 |
|
32 |
[-2, 1, nn.Upsample, [None, 2, 'nearest']],
|
|
|
26 |
|
27 |
# yolov5 head
|
28 |
head:
|
29 |
+
[[-1, 3, Bottleneck, [1024, False]], # 11
|
30 |
[-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1, 0]], # 12 (P5/32-large)
|
31 |
|
32 |
[-2, 1, nn.Upsample, [None, 2, 'nearest']],
|