Metadata-Version: 2.1
Name: openpack-torch
Version: 0.6.0
Summary: PyTorch extention to work around with OpenPack dataset
Home-page: https://open-pack.github.io/home
License: MIT
Author: Yoshimura Naoya
Author-email: yoshimura.naoya@ist.osaka-u.ac.jp
Requires-Python: >=3.10,<3.11
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Requires-Dist: hydra-core (>=1.3.1,<2.0.0)
Requires-Dist: numpy (>=1.22.3,<2.0.0)
Requires-Dist: omegaconf (>=2.3.0,<3.0.0)
Requires-Dist: openpack-toolkit (>=0.7.0,<0.8.0)
Requires-Dist: pandas (>=1.5.2,<2.0.0)
Requires-Dist: pytorch-lightning (>=1.8.6,<2.0.0)
Requires-Dist: torch (>=1.13.1,<2.0.0)
Requires-Dist: tqdm (>=4.64.1,<5.0.0)
Project-URL: Repository, https://github.com/open-pack/openpack-torch
Description-Content-Type: text/markdown

# openpack-torch

[![Test](https://github.com/open-pack/openpack-torch/actions/workflows/test.yaml/badge.svg)](https://github.com/open-pack/openpack-torch/actions/workflows/test.yaml)
[![GitHub Pages](https://github.com/open-pack/openpack-torch/actions/workflows/deploy-docs.yaml/badge.svg)](https://github.com/open-pack/openpack-torch/actions/workflows/deploy-docs.yaml)

PyTorch utilities to work around with [OpenPack Dataset](https://open-pack.github.io/).

## Setup

You can install via pip with the following command.

```bash
# Pip
pip install openpack-torch

# Poetry
poetry add  openpack-torch
```

## Docs

- [Dataset Page](https://open-pack.github.io/)
- [API Docs](https://open-pack.github.io/openpack-torch/openpack_torch)
- [PyPI - openpack-torch](https://pypi.org/project/openpack-torch/)

## Examples

### Operation Recognition (Semantic Segmentation)

#### IMU

- Acceleration
  - [U-Net](./examples/unet/)
  - [DeepConvLSTM](./examples/deep-conv-lstm/)

#### Vision

- Keypoints
  - [ST-GCN](./examples/st-gcn)

#### Scores of Baseline Moodel (Preliminary Experiments)

##### Split: Pilot Challenge

| Model                    | F1 (Test Set) | F1 (Submission Set) | Date       | Code |
|--------------------------|---------------|---------------------|------------|------|
| UNet                     | 0.3451        | 0.3747              | 2022-06-28 | [main.py](./examples/unet/main.py) |
| DeepConvLSTM             | 0.7081        | 0.7695              | 2022-06-28 | [main.py](./examples/deep-conv-lstm/main.py) |
| ST-GCN                   | 0.7024        | 0.6106              | 2022-07-07 | [main.py](./examples/st-gcn/main.py) |

NOTE: F1 = F1-measure (macro average)

##### Split: OpenPack Challenge 2022

This split is defined for OpenPack Challenge 2022.

| Model                    | F1 (Test Set) | F1 (Submission Set) | Date       | Code |
|--------------------------|---------------|---------------------|------------|------|
| UNet                     | TBA           | TBA                 | -          | -    |
| DeepConvLSTM             | TBA           | TBA                 | -          | -    |
| ST-GCN                   | TBA           | TBA                 | -          | -    |

## OpenPack Challenge 2022 @ PerCom2023 WS Bird

We are hosting an activity recognition competition, using the OpenPack dataset at a [PerCom 2023 Workshop](https://bio-navigation.jp/bird2023/)!
The task is very simple: Recognize 10 work operations from the OpenPack dataset.
Please join this exciting opportunity. For more information about the competition, click [here](https://open-pack.github.io/challenge2022).

### Tutorials

- [Train a baseline model and make submission file.](./examples/unet/notebooks/U-Net_Train-Model-and-Make-Submission-File.ipynb) [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/open-pack/openpack-torch/blob/main/examples/unet/notebooks/U-Net_Train-Model-and-Make-Submission-File.ipynb)
- Change input modalities. (coming soon ...)

![OpenPack Challenge Logo](./assets/img/OpenPackCHALLENG-black.png)

## LICENCE

This software (openpack-torch) is distributed under [MIT Licence](./LICENSE).
For the license of "OpenPack Dataset", please check [this site (https://open-pack.github.io/)](https://open-pack.github.io/).

