python-yolov5-icevision 6.0-1

2 files changed, 63 insertions, 0 deletions

diff --git a/.SRCINFO b/.SRCINFO
new file mode 100644
index 000000000000..2cf162d3058f
--- /dev/null
+++ b/.SRCINFO
@@ -0,0 +1,26 @@
+pkgbase = python-yolov5-icevision
+	pkgdesc = Family of object detection architectures and models pretrained on the COCO dataset
+	pkgver = 6.0
+	pkgrel = 1
+	url = https://github.com/airctic/yolov5-icevision
+	arch = any
+	license = GPL3
+	makedepends = python-setuptools
+	depends = python-matplotlib
+	depends = python-numpy
+	depends = python-opencv
+	depends = python-pillow
+	depends = python-pyyaml
+	depends = python-requests
+	depends = python-scipy
+	depends = python-pytorch
+	depends = python-torchvision
+	depends = python-tqdm
+	depends = tensorboard
+	depends = python-pandas
+	depends = python-seaborn
+	depends = python-pytorch-thop-git
+	source = python-yolov5-icevision-6.0.tar.gz::https://github.com/airctic/yolov5-icevision/archive/e7ca90c5c9bc4921941fe5e32e1b88c0cf2946b6.zip
+	md5sums = a4b9f2d8d6207bcdc0bc433e8ce6b38b
+
+pkgname = python-yolov5-icevision
diff --git a/PKGBUILD b/PKGBUILD
new file mode 100644
index 000000000000..f45c3d1c9c7b
--- /dev/null
+++ b/PKGBUILD
@@ -0,0 +1,37 @@
+# Maintainer: piernov <piernov@piernov.org>
+
+pkgname=python-yolov5-icevision
+pkgver=6.0
+pkgrel=1
+_commit=e7ca90c5c9bc4921941fe5e32e1b88c0cf2946b6
+pkgdesc="Family of object detection architectures and models pretrained on the COCO dataset"
+arch=('any')
+url="https://github.com/airctic/yolov5-icevision"
+license=('GPL3')
+depends=('python-matplotlib'
+	'python-numpy'
+	'python-opencv'
+	'python-pillow'
+	'python-pyyaml'
+	'python-requests'
+	'python-scipy'
+	'python-pytorch'
+	'python-torchvision'
+	'python-tqdm'
+	'tensorboard'
+	'python-pandas'
+	'python-seaborn'
+	'python-pytorch-thop-git')
+makedepends=('python-setuptools')
+source=("$pkgname-$pkgver.tar.gz::https://github.com/airctic/yolov5-icevision/archive/$_commit.zip")
+md5sums=('a4b9f2d8d6207bcdc0bc433e8ce6b38b')
+
+build() {
+  cd "$srcdir"/yolov5-icevision-$_commit
+  python setup.py build
+}
+
+package() {
+  cd "$srcdir"/yolov5-icevision-$_commit
+  python setup.py install --root="$pkgdir"/ --optimize=1
+}