aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorYuri Victorovich <yuri@FreeBSD.org>2024-03-22 09:54:50 +0000
committerYuri Victorovich <yuri@FreeBSD.org>2024-03-22 09:55:15 +0000
commit52affa2f6830dbbf76a3441f7ab36d1d30faf29e (patch)
tree26bfb545700c4618d74c44782a7e62e79fc70d08
parentb3180c409b30119ee92be6b5807a6778a7a739ff (diff)
misc/py-timm-vis: New port: Visualizer for PyTorch image models
-rw-r--r--misc/Makefile1
-rw-r--r--misc/py-timm-vis/Makefile28
-rw-r--r--misc/py-timm-vis/distinfo3
-rw-r--r--misc/py-timm-vis/pkg-descr18
4 files changed, 50 insertions, 0 deletions
diff --git a/misc/Makefile b/misc/Makefile
index ddd33e170078..91403f1fa140 100644
--- a/misc/Makefile
+++ b/misc/Makefile
@@ -500,6 +500,7 @@
SUBDIR += py-tflite-support
SUBDIR += py-thermocouples_reference
SUBDIR += py-timm
+ SUBDIR += py-timm-vis
SUBDIR += py-toil
SUBDIR += py-torch-geometric
SUBDIR += py-torchmetrics
diff --git a/misc/py-timm-vis/Makefile b/misc/py-timm-vis/Makefile
new file mode 100644
index 000000000000..41b1bbbefe70
--- /dev/null
+++ b/misc/py-timm-vis/Makefile
@@ -0,0 +1,28 @@
+PORTNAME= timm-vis
+DISTVERSION= 2.0.0
+CATEGORIES= misc python # machine-learning
+MASTER_SITES= PYPI
+PKGNAMEPREFIX= ${PYTHON_PKGNAMEPREFIX}
+DISTNAME= ${PORTNAME:S/-/_/}-${PORTVERSION}
+
+MAINTAINER= yuri@FreeBSD.org
+COMMENT= Visualizer for PyTorch image models
+WWW= https://github.com/novice03/timm-vis
+
+LICENSE= MIT
+LICENSE_FILE= ${WRKSRC}/LICENSE
+
+BUILD_DEPENDS= ${PY_SETUPTOOLS} \
+ ${PYTHON_PKGNAMEPREFIX}wheel>0:devel/py-wheel@${PY_FLAVOR}
+RUN_DEPENDS= ${PYTHON_PKGNAMEPREFIX}matplotlib>0:math/py-matplotlib@${PY_FLAVOR} \
+ ${PYNUMPY} \
+ ${PYTHON_PKGNAMEPREFIX}pillow>0:graphics/py-pillow@${PY_FLAVOR} \
+ ${PYTHON_PKGNAMEPREFIX}pytorch>0:misc/py-pytorch@${PY_FLAVOR} \
+ ${PYTHON_PKGNAMEPREFIX}torchvision>0:misc/py-torchvision@${PY_FLAVOR}
+
+USES= python
+USE_PYTHON= pep517 autoplist
+
+NO_ARCH= yes
+
+.include <bsd.port.mk>
diff --git a/misc/py-timm-vis/distinfo b/misc/py-timm-vis/distinfo
new file mode 100644
index 000000000000..256b5d7013e1
--- /dev/null
+++ b/misc/py-timm-vis/distinfo
@@ -0,0 +1,3 @@
+TIMESTAMP = 1711099867
+SHA256 (timm_vis-2.0.0.tar.gz) = bdc42c3d6b0cb541cc91148e5d501fb9bfbde96820f8749d9b9586854ff531fc
+SIZE (timm_vis-2.0.0.tar.gz) = 6515
diff --git a/misc/py-timm-vis/pkg-descr b/misc/py-timm-vis/pkg-descr
new file mode 100644
index 000000000000..7aec645681d2
--- /dev/null
+++ b/misc/py-timm-vis/pkg-descr
@@ -0,0 +1,18 @@
+timm-vis is a PyTorch image models visualizer.
+
+Implementation of various visualization techniques for pytorch image
+classifiers. This library can be used to visualize and understand any PyTorch
+image classifier. This is NOT an official PyTorch library, nor is it affiliated
+with Ross Wightman's pytorch-image-models library. details.ipynb has visual
+examples of all methods implemented.
+
+Currently, the following methods are implemented:
+* Filter visualization
+* Activations visualization
+* Maximally activated patches
+* Saliency maps
+* Synthetic image generation
+* Adversarial attacks to fool models
+* Feature inversion
+* Grad-CAM
+* Deep Dream