From 334ea9da0d098db63143a68e2e28d41b31ea90f1 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Mon, 9 Sep 2019 19:15:17 +0200
Subject: [PATCH] updates

---
 Dockerfile | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 89bc72ab..5f773d7f 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -35,6 +35,12 @@ WORKDIR /usr/src/app
 # Copy contents
 COPY . /usr/src/app
 
+# Copy weights
+RUN python3 -c "from utils.google_utils import *; \
+    gdrive_download(id='18xqvs_uwAqfTXp-LJCYLYNHBOcrwbrp0', name='weights/darknet53.conv.74'); \
+    gdrive_download(id='1oPCHKsM2JpM-zgyepQciGli9X0MTsJCO', name='weights/yolov3-spp.weights')"
+
+
 # ---------------------------------------------------  Extras Below  ---------------------------------------------------
 
 # Build container
@@ -49,9 +55,9 @@ COPY . /usr/src/app
 # sudo nvidia-docker run --ipc=host --mount type=bind,source="$(pwd)"/coco,target=/usr/src/coco ultralytics/yolov3:v0 python3 train.py
 # sudo nvidia-docker run --ipc=host --mount type=bind,source="$(pwd)"/coco,target=/usr/src/coco ultralytics/yolov3:v0 python3 train.py --batch-size 64 --accumulate 1 --img-size 320 --arc uFBCE --prebias --epochs 27
 
-# Push container to https://hub.docker.com/u/ultralytics
-# docker push ultralytics/yolov3:v0
-
-# Build and Push
+# Build and Push to https://hub.docker.com/u/ultralytics
 # export tag=ultralytics/yolov3:v0 && sudo docker build -t $tag . && docker push $tag
 
+# Kill all running containers
+# sudo docker kill $(sudo docker ps -q)
+