Add Darknet

ufoym · Oct 13, 2018 · e384038 · e384038
1 parent bc37059
commit e384038
Show file tree

Hide file tree

Showing 17 changed files with 459 additions and 18 deletions.
diff --git a/README.md b/README.md
@@ -145,6 +145,11 @@ _You are now ready to begin your journey._
 caffe version 1.0.0
 ```
 
+```$ darknet```
+```
+usage: darknet <function>
+```
+
 ```$ th```
 ```
  │  ______             __   |  Torch7
@@ -261,8 +266,9 @@ This may take several minutes as it compiles a few libraries from scratch.
  [cntk](http://cntk.ai)                            | :x:                  | :x:                | :x:                  | :heavy_check_mark:
  [chainer](https://chainer.org)                    | :x:                  | :x:                | :x:                  | :heavy_check_mark:
  [caffe](http://caffe.berkeleyvision.org)          | :heavy_check_mark:   | :heavy_check_mark: | :heavy_check_mark:   | :heavy_check_mark:
- [caffe2](https://caffe2.ai)                        | :x:                  | :x:                | :x:                  | :heavy_check_mark:
+ [caffe2](https://caffe2.ai)                       | :x:                  | :x:                | :x:                  | :heavy_check_mark:
  [torch](http://torch.ch/)                         | :x:                  | :heavy_check_mark: | :heavy_check_mark:   | :heavy_check_mark:
+ [darknet](https://pjreddie.com/darknet/)          | :x:                  | :x:                | :x:                  | :heavy_check_mark:
 
 
 
@@ -275,12 +281,12 @@ This may take several minutes as it compiles a few libraries from scratch.
 
 .                                                  | CUDA 9.0 / Python 3.6                                    | CUDA 9.0 / Python 2.7                    | CPU-only / Python 3.6                       | CPU-only / Python 2.7
 :------------------------------------------------: | :------------------------------------------------------: | :--------------------------------:       | :-----------------------------------------: | :----------------------------------------:
- all-in-one                                        | `latest` `all` `all-py36` `py36-cu90` `all-py36-cu90`    | `all-py27-cu90` `all-py27` `py27-cu90`        | `all-py36-cpu` `all-cpu` `py36-cpu` `cpu`   | `all-py27-cpu` `py27-cpu`
+ all-in-one                                        | `latest` `all` `all-py36` `py36-cu90` `all-py36-cu90`    | `all-py27-cu90` `all-py27` `py27-cu90`   | `all-py36-cpu` `all-cpu` `py36-cpu` `cpu`   | `all-py27-cpu` `py27-cpu`
  all-in-one with jupyter                           | `all-jupyter-py36-cu90` `all-jupyter-py36` `all-jupyter` | `all-py27-jupyter` `py27-jupyter`        | `all-py36-jupyter-cpu` `py36-jupyter-cpu`   | `all-py27-jupyter-cpu` `py27-jupyter-cpu`
  [theano](http://deeplearning.net/software/theano) | `theano-py36-cu90` `theano-py36` `theano`                | `theano-py27-cu90` `theano-py27`         | `theano-py36-cpu` `theano-cpu`              | `theano-py27-cpu`
  [tensorflow](http://www.tensorflow.org)           | `tensorflow-py36-cu90` `tensorflow-py36` `tensorflow`    | `tensorflow-py27-cu90` `tensorflow-py27` | `tensorflow-py36-cpu` `tensorflow-cpu`      | `tensorflow-py27-cpu`
  [sonnet](https://github.com/deepmind/sonnet)      | `sonnet-py36-cu90` `sonnet-py36` `sonnet`                | `sonnet-py27-cu90` `sonnet-py27`         | `sonnet-py36-cpu` `sonnet-cpu`              | `sonnet-py27-cpu`
- [pytorch](http://pytorch.org)                     | `pytorch-py36-cu90` `pytorch-py36` `pytorch`             | `pytorch-py27-cu90` `pytorch-py27`       | `pytorch-py36` `pytorch`                    | `pytorch-py27`
+ [pytorch](http://pytorch.org)                     | `pytorch-py36-cu90` `pytorch-py36` `pytorch`             | `pytorch-py27-cu90` `pytorch-py27`       | `pytorch-py36-cpu` `pytorch-cpu`            | `pytorch-py27-cpu`
  [keras](https://keras.io)                         | `keras-py36-cu90` `keras-py36` `keras`                   | `keras-py27-cu90` `keras-py27`           | `keras-py36-cpu` `keras-cpu`                | `keras-py27-cpu`
  [lasagne](http://lasagne.readthedocs.io)          | `lasagne-py36-cu90` `lasagne-py36` `lasagne`             | `lasagne-py27-cu90` `lasagne-py27`       | `lasagne-py36-cpu` `lasagne-cpu`            | `lasagne-py27-cpu`
  [mxnet](http://mxnet.incubator.apache.org)        | `mxnet-py36-cu90` `mxnet-py36` `mxnet`                   | `mxnet-py27-cu90` `mxnet-py27`           | `mxnet-py36-cpu` `mxnet-cpu`                | `mxnet-py27-cpu`
@@ -289,6 +295,7 @@ This may take several minutes as it compiles a few libraries from scratch.
  [caffe](http://caffe.berkeleyvision.org)          | `caffe-py36-cu90` `caffe-py36` `caffe`                   | `caffe-py27-cu90` `caffe-py27`           | `caffe-py36-cpu` `caffe-cpu`                | `caffe-py27-cpu`
  [caffe2](https://caffe2.ai)                       | `caffe2-py36-cu90` `caffe2-py36` `caffe2`                | `caffe2-py27-cu90` `caffe2-py27`         | `caffe2-py36-cpu` `caffe2-cpu`              | `caffe2-py27-cpu`
  [torch](http://torch.ch/)                         | `torch-cu90` `torch`                                     | `torch-cu90` `torch`                     | `torch-cpu`                                 | `torch-cpu`
+ [darknet](https://pjreddie.com/darknet/)          | `darknet-cu90` `darknet`                                 | `darknet-cu90` `darknet`                 | `darknet-cpu`                               | `darknet-cpu`
 
 ---
 

diff --git a/circle.yml b/circle.yml
@@ -65,6 +65,15 @@ jobs:
             - run: docker push $DOCKER_REPO:sonnet-py36-cpu            
             - run: docker push $DOCKER_REPO:sonnet-cpu
 
+    "darknet-cu90_darknet":
+        machine: true
+        steps:
+            - checkout
+            - run: docker build -t $DOCKER_REPO:darknet-cu90 -t $DOCKER_REPO:darknet -f docker/Dockerfile.darknet-cu90 .
+            - run: docker login -u $DOCKER_USER -p $DOCKER_PASS            
+            - run: docker push $DOCKER_REPO:darknet-cu90            
+            - run: docker push $DOCKER_REPO:darknet
+
     "all-py27-cpu_py27-cpu":
         machine: true
         steps:
@@ -121,6 +130,14 @@ jobs:
             - run: docker push $DOCKER_REPO:caffe-py36            
             - run: docker push $DOCKER_REPO:caffe
 
+    "darknet-cpu":
+        machine: true
+        steps:
+            - checkout
+            - run: docker build -t $DOCKER_REPO:darknet-cpu -f docker/Dockerfile.darknet-cpu .
+            - run: docker login -u $DOCKER_USER -p $DOCKER_PASS            
+            - run: docker push $DOCKER_REPO:darknet-cpu
+
     "mxnet-py27-cpu":
         machine: true
         steps:
@@ -438,6 +455,15 @@ jobs:
             - run: docker login -u $DOCKER_USER -p $DOCKER_PASS            
             - run: docker push $DOCKER_REPO:caffe-py27-cpu
 
+    "pytorch-py36-cpu_pytorch-cpu":
+        machine: true
+        steps:
+            - checkout
+            - run: docker build -t $DOCKER_REPO:pytorch-py36-cpu -t $DOCKER_REPO:pytorch-cpu -f docker/Dockerfile.pytorch-py36-cpu .
+            - run: docker login -u $DOCKER_USER -p $DOCKER_PASS            
+            - run: docker push $DOCKER_REPO:pytorch-py36-cpu            
+            - run: docker push $DOCKER_REPO:pytorch-cpu
+
     "mxnet-py27-cu90_mxnet-py27":
         machine: true
         steps:
@@ -465,6 +491,14 @@ jobs:
             - run: docker push $DOCKER_REPO:sonnet-py27-cu90            
             - run: docker push $DOCKER_REPO:sonnet-py27
 
+    "pytorch-py27-cpu":
+        machine: true
+        steps:
+            - checkout
+            - run: docker build -t $DOCKER_REPO:pytorch-py27-cpu -f docker/Dockerfile.pytorch-py27-cpu .
+            - run: docker login -u $DOCKER_USER -p $DOCKER_PASS            
+            - run: docker push $DOCKER_REPO:pytorch-py27-cpu
+
     "caffe2-py36-cu90_caffe2-py36_caffe2":
         machine: true
         steps:
@@ -486,12 +520,14 @@ workflows:
             - "tensorflow-py36-cpu_tensorflow-cpu"
             - "cntk-py36-cu90_cntk-py36_cntk"
             - "sonnet-py36-cpu_sonnet-cpu"
+            - "darknet-cu90_darknet"
             - "all-py27-cpu_py27-cpu"
             - "tensorflow-py27-cpu"
             - "keras-py36-cu90_keras-py36_keras"
             - "sonnet-py27-cpu"
             - "all-py36-cpu_all-cpu_py36-cpu_cpu"
             - "caffe-py36-cu90_caffe-py36_caffe"
+            - "darknet-cpu"
             - "mxnet-py27-cpu"
             - "lasagne-py36-cpu_lasagne-cpu"
             - "caffe-py27-cu90_caffe-py27"
@@ -527,7 +563,9 @@ workflows:
             - "tensorflow-py36-cu90_tensorflow-py36_tensorflow"
             - "theano-py27-cu90_theano-py27"
             - "caffe-py27-cpu"
+            - "pytorch-py36-cpu_pytorch-cpu"
             - "mxnet-py27-cu90_mxnet-py27"
             - "caffe-py36-cpu_caffe-cpu"
             - "sonnet-py27-cu90_sonnet-py27"
+            - "pytorch-py27-cpu"
             - "caffe2-py36-cu90_caffe2-py36_caffe2"
diff --git a/docker/Dockerfile.all-jupyter-py27-cpu b/docker/Dockerfile.all-jupyter-py27-cpu
@@ -1,6 +1,7 @@
 # ==================================================================
 # module list
 # ------------------------------------------------------------------
+# darknet       latest (git)
 # python        2.7    (apt)
 # torch         latest (git)
 # chainer       latest (pip)
@@ -43,6 +44,20 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \
         vim \
         && \
 
+# ==================================================================
+# darknet
+# ------------------------------------------------------------------
+
+    $GIT_CLONE https://github.com/pjreddie/darknet.git ~/darknet && \
+    cd ~/darknet && \
+    sed -i 's/GPU=0/GPU=0/g' ~/darknet/Makefile && \
+    sed -i 's/CUDNN=0/CUDNN=0/g' ~/darknet/Makefile && \
+    make -j"$(nproc)" && \
+    cp ~/darknet/include/* /usr/local/include && \
+    cp ~/darknet/*.a /usr/local/lib && \
+    cp ~/darknet/*.so /usr/local/lib && \
+    cp ~/darknet/darknet /usr/local/bin && \
+
 # ==================================================================
 # python
 # ------------------------------------------------------------------

diff --git a/docker/Dockerfile.all-jupyter-py27-cu90 b/docker/Dockerfile.all-jupyter-py27-cu90
@@ -1,6 +1,7 @@
 # ==================================================================
 # module list
 # ------------------------------------------------------------------
+# darknet       latest (git)
 # python        2.7    (apt)
 # torch         latest (git)
 # chainer       latest (pip)
@@ -43,6 +44,20 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \
         vim \
         && \
 
+# ==================================================================
+# darknet
+# ------------------------------------------------------------------
+
+    $GIT_CLONE https://github.com/pjreddie/darknet.git ~/darknet && \
+    cd ~/darknet && \
+    sed -i 's/GPU=0/GPU=1/g' ~/darknet/Makefile && \
+    sed -i 's/CUDNN=0/CUDNN=1/g' ~/darknet/Makefile && \
+    make -j"$(nproc)" && \
+    cp ~/darknet/include/* /usr/local/include && \
+    cp ~/darknet/*.a /usr/local/lib && \
+    cp ~/darknet/*.so /usr/local/lib && \
+    cp ~/darknet/darknet /usr/local/bin && \
+
 # ==================================================================
 # python
 # ------------------------------------------------------------------

diff --git a/docker/Dockerfile.all-jupyter-py36-cpu b/docker/Dockerfile.all-jupyter-py36-cpu
@@ -1,6 +1,7 @@
 # ==================================================================
 # module list
 # ------------------------------------------------------------------
+# darknet       latest (git)
 # python        3.6    (apt)
 # torch         latest (git)
 # chainer       latest (pip)
@@ -43,6 +44,20 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \
         vim \
         && \
 
+# ==================================================================
+# darknet
+# ------------------------------------------------------------------
+
+    $GIT_CLONE https://github.com/pjreddie/darknet.git ~/darknet && \
+    cd ~/darknet && \
+    sed -i 's/GPU=0/GPU=0/g' ~/darknet/Makefile && \
+    sed -i 's/CUDNN=0/CUDNN=0/g' ~/darknet/Makefile && \
+    make -j"$(nproc)" && \
+    cp ~/darknet/include/* /usr/local/include && \
+    cp ~/darknet/*.a /usr/local/lib && \
+    cp ~/darknet/*.so /usr/local/lib && \
+    cp ~/darknet/darknet /usr/local/bin && \
+
 # ==================================================================
 # python
 # ------------------------------------------------------------------

diff --git a/docker/Dockerfile.all-jupyter-py36-cu90 b/docker/Dockerfile.all-jupyter-py36-cu90
@@ -1,6 +1,7 @@
 # ==================================================================
 # module list
 # ------------------------------------------------------------------
+# darknet       latest (git)
 # python        3.6    (apt)
 # torch         latest (git)
 # chainer       latest (pip)
@@ -43,6 +44,20 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \
         vim \
         && \
 
+# ==================================================================
+# darknet
+# ------------------------------------------------------------------
+
+    $GIT_CLONE https://github.com/pjreddie/darknet.git ~/darknet && \
+    cd ~/darknet && \
+    sed -i 's/GPU=0/GPU=1/g' ~/darknet/Makefile && \
+    sed -i 's/CUDNN=0/CUDNN=1/g' ~/darknet/Makefile && \
+    make -j"$(nproc)" && \
+    cp ~/darknet/include/* /usr/local/include && \
+    cp ~/darknet/*.a /usr/local/lib && \
+    cp ~/darknet/*.so /usr/local/lib && \
+    cp ~/darknet/darknet /usr/local/bin && \
+
 # ==================================================================
 # python
 # ------------------------------------------------------------------

diff --git a/docker/Dockerfile.all-py27-cpu b/docker/Dockerfile.all-py27-cpu
@@ -1,6 +1,7 @@
 # ==================================================================
 # module list
 # ------------------------------------------------------------------
+# darknet       latest (git)
 # python        2.7    (apt)
 # torch         latest (git)
 # chainer       latest (pip)
@@ -42,6 +43,20 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \
         vim \
         && \
 
+# ==================================================================
+# darknet
+# ------------------------------------------------------------------
+
+    $GIT_CLONE https://github.com/pjreddie/darknet.git ~/darknet && \
+    cd ~/darknet && \
+    sed -i 's/GPU=0/GPU=0/g' ~/darknet/Makefile && \
+    sed -i 's/CUDNN=0/CUDNN=0/g' ~/darknet/Makefile && \
+    make -j"$(nproc)" && \
+    cp ~/darknet/include/* /usr/local/include && \
+    cp ~/darknet/*.a /usr/local/lib && \
+    cp ~/darknet/*.so /usr/local/lib && \
+    cp ~/darknet/darknet /usr/local/bin && \
+
 # ==================================================================
 # python
 # ------------------------------------------------------------------

diff --git a/docker/Dockerfile.all-py27-cu90 b/docker/Dockerfile.all-py27-cu90
@@ -1,6 +1,7 @@
 # ==================================================================
 # module list
 # ------------------------------------------------------------------
+# darknet       latest (git)
 # python        2.7    (apt)
 # torch         latest (git)
 # chainer       latest (pip)
@@ -42,6 +43,20 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \
         vim \
         && \
 
+# ==================================================================
+# darknet
+# ------------------------------------------------------------------
+
+    $GIT_CLONE https://github.com/pjreddie/darknet.git ~/darknet && \
+    cd ~/darknet && \
+    sed -i 's/GPU=0/GPU=1/g' ~/darknet/Makefile && \
+    sed -i 's/CUDNN=0/CUDNN=1/g' ~/darknet/Makefile && \
+    make -j"$(nproc)" && \
+    cp ~/darknet/include/* /usr/local/include && \
+    cp ~/darknet/*.a /usr/local/lib && \
+    cp ~/darknet/*.so /usr/local/lib && \
+    cp ~/darknet/darknet /usr/local/bin && \
+
 # ==================================================================
 # python
 # ------------------------------------------------------------------

diff --git a/docker/Dockerfile.all-py36-cpu b/docker/Dockerfile.all-py36-cpu
@@ -1,6 +1,7 @@
 # ==================================================================
 # module list
 # ------------------------------------------------------------------
+# darknet       latest (git)
 # python        3.6    (apt)
 # torch         latest (git)
 # chainer       latest (pip)
@@ -42,6 +43,20 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \
         vim \
         && \
 
+# ==================================================================
+# darknet
+# ------------------------------------------------------------------
+
+    $GIT_CLONE https://github.com/pjreddie/darknet.git ~/darknet && \
+    cd ~/darknet && \
+    sed -i 's/GPU=0/GPU=0/g' ~/darknet/Makefile && \
+    sed -i 's/CUDNN=0/CUDNN=0/g' ~/darknet/Makefile && \
+    make -j"$(nproc)" && \
+    cp ~/darknet/include/* /usr/local/include && \
+    cp ~/darknet/*.a /usr/local/lib && \
+    cp ~/darknet/*.so /usr/local/lib && \
+    cp ~/darknet/darknet /usr/local/bin && \
+
 # ==================================================================
 # python
 # ------------------------------------------------------------------

diff --git a/docker/Dockerfile.all-py36-cu90 b/docker/Dockerfile.all-py36-cu90
@@ -1,6 +1,7 @@
 # ==================================================================
 # module list
 # ------------------------------------------------------------------
+# darknet       latest (git)
 # python        3.6    (apt)
 # torch         latest (git)
 # chainer       latest (pip)
@@ -42,6 +43,20 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \
         vim \
         && \
 
+# ==================================================================
+# darknet
+# ------------------------------------------------------------------
+
+    $GIT_CLONE https://github.com/pjreddie/darknet.git ~/darknet && \
+    cd ~/darknet && \
+    sed -i 's/GPU=0/GPU=1/g' ~/darknet/Makefile && \
+    sed -i 's/CUDNN=0/CUDNN=1/g' ~/darknet/Makefile && \
+    make -j"$(nproc)" && \
+    cp ~/darknet/include/* /usr/local/include && \
+    cp ~/darknet/*.a /usr/local/lib && \
+    cp ~/darknet/*.so /usr/local/lib && \
+    cp ~/darknet/darknet /usr/local/bin && \
+
 # ==================================================================
 # python
 # ------------------------------------------------------------------