You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
111 lines
3.5 KiB
111 lines
3.5 KiB
# Maintainer : Daniel Bermond < gmail-com: danielbermond >
|
|
# Contributor: Micah Chambers <micahc.vt@gmail.com>
|
|
|
|
pkgname=caffe
|
|
pkgver=1.0
|
|
pkgrel=11
|
|
pkgdesc='A deep learning framework made with expression, speed, and modularity in mind (cpu only)'
|
|
arch=('i686' 'x86_64')
|
|
url='http://caffe.berkeleyvision.org/'
|
|
license=('BSD')
|
|
depends=(
|
|
# official repositories:
|
|
'openblas' 'lapack' 'boost-libs' 'protobuf' 'google-glog' 'gflags'
|
|
'hdf5' 'opencv' 'leveldb' 'lmdb' 'python' 'cython'
|
|
'python-numpy' 'python-scipy' 'python-matplotlib' 'ipython'
|
|
'python-h5py' 'python-networkx' 'python-nose' 'python-pandas'
|
|
'python-dateutil' 'python-protobuf' 'python-gflags' 'python-yaml'
|
|
'python-pillow' 'python-six'
|
|
# AUR:
|
|
'python-leveldb' 'python-scikit-image' 'python-pydotplus'
|
|
# NOTE:
|
|
# python-pydotplus (or python-pydot) is required by python executable 'draw_net.py'
|
|
# https://github.com/BVLC/caffe/blob/1.0/python/caffe/draw.py#L7-L22
|
|
)
|
|
makedepends=('boost' 'doxygen' 'texlive-core')
|
|
replaces=('caffe-cpu')
|
|
source=("${pkgname}-${pkgver}.tar.gz"::"https://github.com/BVLC/caffe/archive/${pkgver}.tar.gz"
|
|
'Makefile.config')
|
|
sha256sums=('71d3c9eb8a183150f965a465824d01fe82826c22505f7aa314f700ace03fa77f'
|
|
'9cbe8ea31d70904ec02a4ca3978aa072c99f10aff0629e7616c51bf4a6ca92f5')
|
|
|
|
prepare() {
|
|
cd "${pkgname}-${pkgver}"
|
|
|
|
local _pythonver
|
|
local _pythonmaj
|
|
local _opencvmaj
|
|
|
|
_pythonver="$(python --version | awk '{ print $2 }' | grep -o '^[0-9]*\.[0-9]*')"
|
|
_pythonmaj="$(python --version | awk '{ print $2 }' | awk -F'.' '{ print $1 }')"
|
|
_opencvmaj="$(opencv_version | awk -F'.' '{ print $1 }')"
|
|
|
|
# copy configuration options
|
|
cp -af "${srcdir}/Makefile.config" .
|
|
|
|
# make sure to use the correct versions of python and opencv
|
|
|
|
if ! grep -q "python${_pythonver}" Makefile.config
|
|
then
|
|
sed -i "s/python[0-9]*\.[0-9]*/python${_pythonver}/" Makefile.config
|
|
fi
|
|
|
|
if ! grep -q "boost_python${_pythonmaj}" Makefile.config
|
|
then
|
|
sed -i "/boost_python[0-9]/s/[0-9]/${_pythonmaj}/" Makefile.config
|
|
fi
|
|
|
|
if ! grep -q "OPENCV_VERSION[[:space:]]:=[[:space:]]${_opencvmaj}" Makefile.config
|
|
then
|
|
sed -i "/OPENCV_VERSION/s/[0-9]*$/${_opencvmaj}/" Makefile.config
|
|
fi
|
|
}
|
|
|
|
build() {
|
|
cd "${pkgname}-${pkgver}"
|
|
|
|
make all pycaffe
|
|
rm -rf doxygen
|
|
make docs distribute
|
|
}
|
|
|
|
# uncomment this block if you want to run the checks/tests
|
|
#check() {
|
|
# cd "${pkgname}-${pkgver}"
|
|
# make test
|
|
# make runtest
|
|
#}
|
|
|
|
package() {
|
|
cd "${pkgname}-${pkgver}/distribute"
|
|
|
|
local _pythonver
|
|
_pythonver="$(python --version | awk '{ print $2 }' | grep -o '^[0-9]*\.[0-9]*')"
|
|
|
|
mkdir -p "$pkgdir"/usr/{bin,include,lib/python"$_pythonver"/site-packages,share/doc}
|
|
|
|
# binaries
|
|
install -m755 bin/* "${pkgdir}/usr/bin"
|
|
|
|
# library
|
|
cp -a lib/libcaffe.so* "${pkgdir}/usr/lib"
|
|
chmod 755 "${pkgdir}/usr/lib"/libcaffe.so.*.*.*
|
|
|
|
# headers
|
|
cp -a include "${pkgdir}/usr"
|
|
|
|
# python
|
|
install -m755 python/*.py "${pkgdir}/usr/bin"
|
|
cp -a python/caffe "${pkgdir}/usr/lib/python${_pythonver}/site-packages"
|
|
|
|
# proto
|
|
install -D -m644 proto/caffe.proto -t "${pkgdir}/usr/share/caffe"
|
|
|
|
cd "${srcdir}/${pkgname}-${pkgver}"
|
|
|
|
# docs
|
|
cp -a doxygen/html "${pkgdir}/usr/share/doc/${pkgname}"
|
|
|
|
# license
|
|
install -D -m644 LICENSE -t "${pkgdir}/usr/share/licenses/${pkgname}"
|
|
}
|
|
|