PKGBUILD file for darknet-alexeyab

PKGBUILD-darknet-alexeyab-yolov4-pre

# Maintainer: acxz <akashpatel2008 at yahoo dot com>
pkgname=darknet-alexeyab
pkgver=r1833.bef28445
pkgrel=1
pkgdesc='YOLO: Real Time Object Detection Neural Network Library (AlexeyAB fork)'
arch=('i686' 'x86_64')
url='https://github.com/AlexeyAB/darknet'
license=('YOLO')
depends=()
optdepends=(opencv)
makedepends=(git cmake cuda-10.2 cudnn7-cuda10.2)
_pkgname=darknet
provides=('darknet')
conflicts=()
source=("${_pkgname}::git+https://github.com/AlexeyAB/darknet.git#tag=darknet_yolo_v4_pre"
        LICENSE)
md5sums=('SKIP'
         '4714f70f7f315d04508e3fd63d9b9232')

pkgver() {
  cd "$_pkgname"
  printf "r%s.%s" "$(git rev-list --count HEAD)" "$(git rev-parse --short HEAD)"
}

build() {
  mkdir -p "${srcdir}/${_pkgname}/build-release"
  cd "${srcdir}/${_pkgname}/build-release"

  cmake .. \
        -DENABLE_OPENCV=OFF \
        -DCMAKE_BUILD_TYPE="Release" \
        -DCUDA_TOOLKIT_ROOT_DIR=/opt/cuda-10.2 \
        -DCUDNN_INCLUDE_DIR=/opt/cudnn7-cuda10.2/include \
        -DCUDNN_LIBRARY=/opt/cudnn7-cuda10.2/lib/libcudnn.so

  make
}

package() {
  msg "Installing files"

  cd "${srcdir}/${_pkgname}/build-release/"
  #make DESTDIR="${pkgdir}/" install

  # Create usr directory
  mkdir $pkgdir/usr

  # bin
  mkdir $pkgdir/usr/bin

  # Check if opencv is installed
  if (pacman -Qqs opencv >/dev/null) ; then
    cp $srcdir/${_pkgname}/build-release/{darknet,uselib} $pkgdir/usr/bin/
  else
    cp $srcdir/${_pkgname}/build-release/darknet $pkgdir/usr/bin/
  fi

  # include
  mkdir -p $pkgdir/usr/include/darknet
  cp -r $srcdir/${_pkgname}/include/{darknet.h,yolo_v2_class.hpp} $pkgdir/usr/include/darknet

  # lib
  mkdir $pkgdir/usr/lib
  cp $srcdir/${_pkgname}/build-release/libdark.so $pkgdir/usr/lib/

  chown -R root:root $pkgdir/usr
  chmod -R 755 $pkgdir/usr
}