deepethogram 0.1.4__tar.gz → 0.3.0__tar.gz

{deepethogram-0.1.4 → deepethogram-0.3.0}/MANIFEST.in

@@ -1,3 +1,3 @@
 include README.md
 include deepethogram/gui/icons/*.png
-recursive-include deepethogram/conf *
+recursive-include deepethogram/conf *

{deepethogram-0.1.4/deepethogram.egg-info → deepethogram-0.3.0}/PKG-INFO

@@ -1,20 +1,35 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: deepethogram
-Version: 0.1.4
+Version: 0.3.0
 Summary: Temporal action detection for biology
-Author: Jim Bohnslav
-Author-email: jbohnslav@gmail.com
+Author-email: Jim Bohnslav <jbohnslav@gmail.com>
 Classifier: Programming Language :: Python :: 3
 Classifier: Operating System :: OS Independent
-Requires-Python: >=3.6
+Requires-Python: <3.12,>=3.9
 Description-Content-Type: text/markdown
+Requires-Dist: chardet
+Requires-Dist: h5py
+Requires-Dist: kornia>=0.5
+Requires-Dist: matplotlib
+Requires-Dist: numpy
+Requires-Dist: omegaconf>=2
+Requires-Dist: opencv-python-headless
+Requires-Dist: opencv-transforms
+Requires-Dist: pandas<2.1,>=1.5
+Requires-Dist: PySide6>=6.6.0
+Requires-Dist: scikit-learn<1.4,>=1.2
+Requires-Dist: scipy<1.11,>=1.9
+Requires-Dist: torch==2.4.1
+Requires-Dist: tqdm
+Requires-Dist: vidio
+Requires-Dist: pytorch_lightning==1.6.5
 
 # DeepEthogram
 - Written by Jim Bohnslav, except where as noted
 - JBohnslav@gmail.com
 
-DeepEthogram is an open-source package for automatically classifying each frame of a video into a set of pre-defined 
-behaviors. Designed for neuroscience research, it could be used in any scenario where you need to detect actions from 
+DeepEthogram is an open-source package for automatically classifying each frame of a video into a set of pre-defined
+behaviors. Designed for neuroscience research, it could be used in any scenario where you need to detect actions from
 each frame of a video.
 
 Example use cases:
@@ -23,7 +38,7 @@ Example use cases:
 * Counting licks from video for appetite measurement
 * Measuring reach onset times for alignment with neural activity
 
-DeepEthogram uses state-of-the-art algorithms for *temporal action detection*. We build on the following previous machine 
+DeepEthogram uses state-of-the-art algorithms for *temporal action detection*. We build on the following previous machine
 learning research into action detection:
 * [Hidden Two-Stream Convolutional Networks for Action Recognition](https://arxiv.org/abs/1704.00389)
 * [Temporal Gaussian Mixture Layer for Videos](https://arxiv.org/abs/1803.06316)
@@ -31,24 +46,46 @@ learning research into action detection:
 ![deepethogram schematic](docs/images/deepethogram_schematic.png)
 
 ## Installation
-For full installation instructions, see [this readme file](docs/installation.md). 
+DeepEthogram recently migrated from Miniconda/pip to [uv](https://docs.astral.sh/uv/). Current releases are uv-first,
+recommend Python 3.11, support Python `>=3.9,<3.12`, and use PySide6 for the GUI.
 
-In brief: 
-* [Install PyTorch](https://pytorch.org/) 
-* `pip install deepethogram`
+Install uv first:
+
+```bash
+curl -LsSf https://astral.sh/uv/install.sh | sh
+# or
+brew install uv
+```
+
+For development from this repository:
+
+```bash
+git clone https://github.com/jbohnslav/deepethogram.git
+cd deepethogram
+uv sync
+uv run deepethogram
+```
+
+For a user install:
+
+```bash
+uv pip install deepethogram
+```
+
+Detailed setup, troubleshooting, and legacy conda/pip instructions for older releases are in [docs/installation.md](docs/installation.md).
 
 ## Data
-**NEW!** All datasets collected and annotated by the DeepEthogram authors are now available from this DropBox link: 
+**NEW!** All datasets collected and annotated by the DeepEthogram authors are now available from this DropBox link:
 https://www.dropbox.com/sh/3lilfob0sz21och/AABv8o8KhhRQhYCMNu0ilR8wa?dl=0
 
-If you have issues downloading the data, please raise an issue on Github. 
+If you have issues downloading the data, please raise an issue on Github.
 
 ## COLAB
-I've written a Colab notebook that shows how to upload your data and train models. You can also use this if you don't 
-have access to a decent GPU. 
+I've written a Colab notebook that shows how to upload your data and train models. You can also use this if you don't
+have access to a decent GPU.
 
-To use it, please [click this link to the Colab notebook](https://colab.research.google.com/drive/1Nf9FU7FD77wgvbUFc608839v2jPYgDhd?usp=sharing). 
-Then, click `copy to Drive` at the top. You won't be able to save your changes to the notebook as-is. 
+To use it, please [click this link to the Colab notebook](https://colab.research.google.com/drive/1Nf9FU7FD77wgvbUFc608839v2jPYgDhd?usp=sharing).
+Then, click `copy to Drive` at the top. You won't be able to save your changes to the notebook as-is.
 
 
 ## News
@@ -56,27 +93,27 @@ We now support docker! Docker is a way to run `deepethogram` in completely repro
 with other system dependencies. [See docs/Docker for more information](docs/docker.md)
 
 ## Pretrained models
-Rather than start from scratch, we will start with model weights pretrained on the Kinetics700 dataset. Go to 
+Rather than start from scratch, we will start with model weights pretrained on the Kinetics700 dataset. Go to
 To download the pretrained weights, please use [this Google Drive link](https://drive.google.com/file/d/1ntIZVbOG1UAiFVlsAAuKEBEVCVevyets/view?usp=sharing).
-Unzip the files in your `project/models` directory. Make sure that you don't add an extra directory when unzipping! The path should be 
+Unzip the files in your `project/models` directory. Make sure that you don't add an extra directory when unzipping! The path should be
 `your_project/models/pretrained_models/{models 1:6}`, not `your_project/models/pretrained_models/pretrained_models/{models1:6}`.
 
 ## Licensing
 Copyright (c) 2020 - President and Fellows of Harvard College. All rights reserved.
 
-This software is free for academic use. For commercial use, please contact the Harvard Office of Technology 
-Development (hms_otd@harvard.edu) with cc to Dr. Chris Harvey. For details, see [license.txt](license.txt). 
+This software is free for academic use. For commercial use, please contact the Harvard Office of Technology
+Development (hms_otd@harvard.edu) with cc to Dr. Chris Harvey. For details, see [license.txt](license.txt).
 
 ## Usage
 ### [To use the GUI, click](docs/using_gui.md)
 #### [To use the command line interface, click](docs/using_CLI.md)
 
 ## Dependencies
-The major dependencies for DeepEthogram are as follows: 
+The major dependencies for DeepEthogram are as follows:
 * pytorch, torchvision: all the neural networks, training, and inference pipelines were written in PyTorch
 * pytorch-lightning: for nice model training base classes
 * kornia: for GPU-based image augmentations
-* pyside2: for the GUI
+* PySide6: for the GUI
 * opencv: for video and image reading and writing
 * opencv_transforms: for fast image augmentation
 * scikit-learn, scipy: for binary classification metrics
@@ -87,25 +124,60 @@ The major dependencies for DeepEthogram are as follows:
 * tqdm: for nice progress bars
 
 ## Hardware requirements
-For GUI usage, we expect that the users will be working on a local workstation with a good NVIDIA graphics card. For training via a cluster, you can use the command line interface. 
+For GUI usage, we expect that the users will be working on a local workstation with a good NVIDIA graphics card. For training via a cluster, you can use the command line interface.
 
 * CPU: 4 cores or more for parallel data loading
 * Hard Drive: SSD at minimum, NVMe drive is better.
-* GPU: DeepEthogram speed is directly related to GPU performance. An NVIDIA GPU is absolutely required, as PyTorch uses 
-CUDA, while AMD does not. 
+* GPU: DeepEthogram speed is directly related to GPU performance. An NVIDIA GPU is absolutely required, as PyTorch uses
+CUDA, while AMD does not.
 The more VRAM you have, the more data you can fit in one batch, which generally increases performance. a
 I'd recommend 6GB VRAM at absolute minimum. 8GB is better, with 10+ GB preferred.
-Recommended GPUs: `RTX 3090`, `RTX 3080`, `Titan RTX`, `2080 Ti`, `2080 super`, `2080`, `1080 Ti`, `2070 super`, `2070` 
-Some older ones might also be fine, like a `1080` or even `1070 Ti`/ `1070`. 
+Recommended GPUs: `RTX 3090`, `RTX 3080`, `Titan RTX`, `2080 Ti`, `2080 super`, `2080`, `1080 Ti`, `2070 super`, `2070`
+Some older ones might also be fine, like a `1080` or even `1070 Ti`/ `1070`.
 
 ## testing
-Test coverage is still low, but in the future we will be expanding our unit tests. 
+Test coverage is still low, but in the future we will be expanding our unit tests.
 
 First, download a copy of [`testing_deepethogram_archive.zip`](https://drive.google.com/file/d/1IFz4ABXppVxyuhYik8j38k9-Fl9kYKHo/view?usp=sharing)
- Make a directory in tests called `DATA`. Unzip this and move it to the `deepethogram/tests/DATA` 
-directory, so that the path is `deepethogram/tests/DATA/testing_deepethogram_archive/{DATA,models,project_config.yaml}`. Then run `pytest tests/` to run. 
-the `zz_commandline` test module will take a few minutes, as it is an end-to-end test that performs model training 
-and inference. Its name reflects the fact that it should come last in testing. 
+Make a directory in tests called `DATA`. Unzip this and move it to the `deepethogram/tests/DATA`
+directory, so that the path is `deepethogram/tests/DATA/testing_deepethogram_archive/{DATA,models,project_config.yaml}`.
+
+To run tests:
+```bash
+# Run all tests except GPU tests (default)
+uv run pytest tests/
+
+# Run only GPU tests (requires NVIDIA GPU)
+uv run pytest -m gpu
+
+# Run all tests including GPU tests
+uv run pytest -m ""
+```
+
+GPU tests are skipped by default as they require significant computational resources and time to complete. These tests perform end-to-end model training and inference.
+
+## Developer Guide
+### Code Style and Pre-commit Hooks
+We use pre-commit hooks to maintain code quality and consistency. The hooks include:
+- Ruff for Python linting and formatting
+- Various file checks (trailing whitespace, YAML validation, etc.)
+
+To set up the development environment:
+
+1. Install the development dependencies:
+```bash
+uv sync --dev
+```
+
+2. Install pre-commit hooks:
+```bash
+uvx pre-commit install
+```
+
+The hooks will run automatically on every commit. You can also run them manually on all files:
+```bash
+uvx pre-commit run --all-files
+```
 
 ## Changelog
 * 0.1.4: bugfixes for dependencies; added docker
@@ -113,6 +185,6 @@ and inference. Its name reflects the fact that it should come last in testing.
 * 0.1.1.post1/2: batch prediction
 * 0.1.1.post0: flow generator metric bug fix
 * 0.1.1: bug fixes
-* 0.1: deepethogram beta! See above for details. 
+* 0.1: deepethogram beta! See above for details.
 * 0.0.1.post1: bug fixes and video conversion scripts added
 * 0.0.1: initial version

deepethogram-0.1.4/PKG-INFO → deepethogram-0.3.0/README.md

@@ -1,20 +1,9 @@
-Metadata-Version: 2.1
-Name: deepethogram
-Version: 0.1.4
-Summary: Temporal action detection for biology
-Author: Jim Bohnslav
-Author-email: jbohnslav@gmail.com
-Classifier: Programming Language :: Python :: 3
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.6
-Description-Content-Type: text/markdown
-
 # DeepEthogram
 - Written by Jim Bohnslav, except where as noted
 - JBohnslav@gmail.com
 
-DeepEthogram is an open-source package for automatically classifying each frame of a video into a set of pre-defined 
-behaviors. Designed for neuroscience research, it could be used in any scenario where you need to detect actions from 
+DeepEthogram is an open-source package for automatically classifying each frame of a video into a set of pre-defined
+behaviors. Designed for neuroscience research, it could be used in any scenario where you need to detect actions from
 each frame of a video.
 
 Example use cases:
@@ -23,7 +12,7 @@ Example use cases:
 * Counting licks from video for appetite measurement
 * Measuring reach onset times for alignment with neural activity
 
-DeepEthogram uses state-of-the-art algorithms for *temporal action detection*. We build on the following previous machine 
+DeepEthogram uses state-of-the-art algorithms for *temporal action detection*. We build on the following previous machine
 learning research into action detection:
 * [Hidden Two-Stream Convolutional Networks for Action Recognition](https://arxiv.org/abs/1704.00389)
 * [Temporal Gaussian Mixture Layer for Videos](https://arxiv.org/abs/1803.06316)
@@ -31,24 +20,46 @@ learning research into action detection:
 ![deepethogram schematic](docs/images/deepethogram_schematic.png)
 
 ## Installation
-For full installation instructions, see [this readme file](docs/installation.md). 
+DeepEthogram recently migrated from Miniconda/pip to [uv](https://docs.astral.sh/uv/). Current releases are uv-first,
+recommend Python 3.11, support Python `>=3.9,<3.12`, and use PySide6 for the GUI.
+
+Install uv first:
+
+```bash
+curl -LsSf https://astral.sh/uv/install.sh | sh
+# or
+brew install uv
+```
+
+For development from this repository:
 
-In brief: 
-* [Install PyTorch](https://pytorch.org/) 
-* `pip install deepethogram`
+```bash
+git clone https://github.com/jbohnslav/deepethogram.git
+cd deepethogram
+uv sync
+uv run deepethogram
+```
+
+For a user install:
+
+```bash
+uv pip install deepethogram
+```
+
+Detailed setup, troubleshooting, and legacy conda/pip instructions for older releases are in [docs/installation.md](docs/installation.md).
 
 ## Data
-**NEW!** All datasets collected and annotated by the DeepEthogram authors are now available from this DropBox link: 
+**NEW!** All datasets collected and annotated by the DeepEthogram authors are now available from this DropBox link:
 https://www.dropbox.com/sh/3lilfob0sz21och/AABv8o8KhhRQhYCMNu0ilR8wa?dl=0
 
-If you have issues downloading the data, please raise an issue on Github. 
+If you have issues downloading the data, please raise an issue on Github.
 
 ## COLAB
-I've written a Colab notebook that shows how to upload your data and train models. You can also use this if you don't 
-have access to a decent GPU. 
+I've written a Colab notebook that shows how to upload your data and train models. You can also use this if you don't
+have access to a decent GPU.
 
-To use it, please [click this link to the Colab notebook](https://colab.research.google.com/drive/1Nf9FU7FD77wgvbUFc608839v2jPYgDhd?usp=sharing). 
-Then, click `copy to Drive` at the top. You won't be able to save your changes to the notebook as-is. 
+To use it, please [click this link to the Colab notebook](https://colab.research.google.com/drive/1Nf9FU7FD77wgvbUFc608839v2jPYgDhd?usp=sharing).
+Then, click `copy to Drive` at the top. You won't be able to save your changes to the notebook as-is.
 
 
 ## News
@@ -56,27 +67,27 @@ We now support docker! Docker is a way to run `deepethogram` in completely repro
 with other system dependencies. [See docs/Docker for more information](docs/docker.md)
 
 ## Pretrained models
-Rather than start from scratch, we will start with model weights pretrained on the Kinetics700 dataset. Go to 
+Rather than start from scratch, we will start with model weights pretrained on the Kinetics700 dataset. Go to
 To download the pretrained weights, please use [this Google Drive link](https://drive.google.com/file/d/1ntIZVbOG1UAiFVlsAAuKEBEVCVevyets/view?usp=sharing).
-Unzip the files in your `project/models` directory. Make sure that you don't add an extra directory when unzipping! The path should be 
+Unzip the files in your `project/models` directory. Make sure that you don't add an extra directory when unzipping! The path should be
 `your_project/models/pretrained_models/{models 1:6}`, not `your_project/models/pretrained_models/pretrained_models/{models1:6}`.
 
 ## Licensing
 Copyright (c) 2020 - President and Fellows of Harvard College. All rights reserved.
 
-This software is free for academic use. For commercial use, please contact the Harvard Office of Technology 
-Development (hms_otd@harvard.edu) with cc to Dr. Chris Harvey. For details, see [license.txt](license.txt). 
+This software is free for academic use. For commercial use, please contact the Harvard Office of Technology
+Development (hms_otd@harvard.edu) with cc to Dr. Chris Harvey. For details, see [license.txt](license.txt).
 
 ## Usage
 ### [To use the GUI, click](docs/using_gui.md)
 #### [To use the command line interface, click](docs/using_CLI.md)
 
 ## Dependencies
-The major dependencies for DeepEthogram are as follows: 
+The major dependencies for DeepEthogram are as follows:
 * pytorch, torchvision: all the neural networks, training, and inference pipelines were written in PyTorch
 * pytorch-lightning: for nice model training base classes
 * kornia: for GPU-based image augmentations
-* pyside2: for the GUI
+* PySide6: for the GUI
 * opencv: for video and image reading and writing
 * opencv_transforms: for fast image augmentation
 * scikit-learn, scipy: for binary classification metrics
@@ -87,25 +98,60 @@ The major dependencies for DeepEthogram are as follows:
 * tqdm: for nice progress bars
 
 ## Hardware requirements
-For GUI usage, we expect that the users will be working on a local workstation with a good NVIDIA graphics card. For training via a cluster, you can use the command line interface. 
+For GUI usage, we expect that the users will be working on a local workstation with a good NVIDIA graphics card. For training via a cluster, you can use the command line interface.
 
 * CPU: 4 cores or more for parallel data loading
 * Hard Drive: SSD at minimum, NVMe drive is better.
-* GPU: DeepEthogram speed is directly related to GPU performance. An NVIDIA GPU is absolutely required, as PyTorch uses 
-CUDA, while AMD does not. 
+* GPU: DeepEthogram speed is directly related to GPU performance. An NVIDIA GPU is absolutely required, as PyTorch uses
+CUDA, while AMD does not.
 The more VRAM you have, the more data you can fit in one batch, which generally increases performance. a
 I'd recommend 6GB VRAM at absolute minimum. 8GB is better, with 10+ GB preferred.
-Recommended GPUs: `RTX 3090`, `RTX 3080`, `Titan RTX`, `2080 Ti`, `2080 super`, `2080`, `1080 Ti`, `2070 super`, `2070` 
-Some older ones might also be fine, like a `1080` or even `1070 Ti`/ `1070`. 
+Recommended GPUs: `RTX 3090`, `RTX 3080`, `Titan RTX`, `2080 Ti`, `2080 super`, `2080`, `1080 Ti`, `2070 super`, `2070`
+Some older ones might also be fine, like a `1080` or even `1070 Ti`/ `1070`.
 
 ## testing
-Test coverage is still low, but in the future we will be expanding our unit tests. 
+Test coverage is still low, but in the future we will be expanding our unit tests.
 
 First, download a copy of [`testing_deepethogram_archive.zip`](https://drive.google.com/file/d/1IFz4ABXppVxyuhYik8j38k9-Fl9kYKHo/view?usp=sharing)
- Make a directory in tests called `DATA`. Unzip this and move it to the `deepethogram/tests/DATA` 
-directory, so that the path is `deepethogram/tests/DATA/testing_deepethogram_archive/{DATA,models,project_config.yaml}`. Then run `pytest tests/` to run. 
-the `zz_commandline` test module will take a few minutes, as it is an end-to-end test that performs model training 
-and inference. Its name reflects the fact that it should come last in testing. 
+Make a directory in tests called `DATA`. Unzip this and move it to the `deepethogram/tests/DATA`
+directory, so that the path is `deepethogram/tests/DATA/testing_deepethogram_archive/{DATA,models,project_config.yaml}`.
+
+To run tests:
+```bash
+# Run all tests except GPU tests (default)
+uv run pytest tests/
+
+# Run only GPU tests (requires NVIDIA GPU)
+uv run pytest -m gpu
+
+# Run all tests including GPU tests
+uv run pytest -m ""
+```
+
+GPU tests are skipped by default as they require significant computational resources and time to complete. These tests perform end-to-end model training and inference.
+
+## Developer Guide
+### Code Style and Pre-commit Hooks
+We use pre-commit hooks to maintain code quality and consistency. The hooks include:
+- Ruff for Python linting and formatting
+- Various file checks (trailing whitespace, YAML validation, etc.)
+
+To set up the development environment:
+
+1. Install the development dependencies:
+```bash
+uv sync --dev
+```
+
+2. Install pre-commit hooks:
+```bash
+uvx pre-commit install
+```
+
+The hooks will run automatically on every commit. You can also run them manually on all files:
+```bash
+uvx pre-commit run --all-files
+```
 
 ## Changelog
 * 0.1.4: bugfixes for dependencies; added docker
@@ -113,6 +159,6 @@ and inference. Its name reflects the fact that it should come last in testing.
 * 0.1.1.post1/2: batch prediction
 * 0.1.1.post0: flow generator metric bug fix
 * 0.1.1: bug fixes
-* 0.1: deepethogram beta! See above for details. 
+* 0.1: deepethogram beta! See above for details.
 * 0.0.1.post1: bug fixes and video conversion scripts added
 * 0.0.1: initial version

{deepethogram-0.1.4 → deepethogram-0.3.0}/deepethogram/__init__.py

@@ -7,9 +7,12 @@
 # from deepethogram.sequence.inference import sequence_inference
 import importlib.util
 
-spec = importlib.util.find_spec('hydra')
+spec = importlib.util.find_spec("hydra")
 if spec is not None:
-    raise ValueError('Hydra installation found. Please run pip uninstall hydra-core: {}'.format(spec))
+    raise ValueError(
+        "Hydra installation found. Remove it from the current environment, for example with "
+        "`uv pip uninstall hydra-core`: {}".format(spec)
+    )
 # try:
 #     import hydra
 # except Exception as e:

{deepethogram-0.1.4 → deepethogram-0.3.0}/deepethogram/__main__.py

@@ -6,5 +6,5 @@ from deepethogram.gui.main import entry
 # def main(cfg: DictConfig) -> None:
 #     run(cfg)
 
-if __name__ == '__main__':
-    entry()
+if __name__ == "__main__":
+    entry()