summarylogtreecommitdiffstats
path: root/PKGBUILD
blob: b6f558b3819a97549697dbaf625089e57327a62c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
# Maintainer: piernov <piernov@piernov.org>

pkgname=python-segmentation-models-pytorch
pkgver=0.3.2
pkgrel=1
pkgdesc="Neural Networks for Image Segmentation based on PyTorch"
arch=('any')
url="https://github.com/qubvel/segmentation_models.pytorch"
license=('MIT')
depends=('python-pytorch' 'python-pretrainedmodels' 'python-efficientnet-pytorch' 'python-timm' 'python-tqdm' 'python-pillow')
makedepends=('python-build' 'python-installer' 'python-wheel' 'python-setuptools-scm')
source=("git+https://github.com/qubvel/segmentation_models.pytorch.git#tag=v$pkgver")
sha256sums=('SKIP')

build() {
  cd "$srcdir"/segmentation_models.pytorch
  python -m build --wheel --no-isolation
}

package() {
  cd "$srcdir"/segmentation_models.pytorch
  python -m installer --destdir="$pkgdir" dist/*.whl
}