Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/unity-technologies/pysolotools
Python toolchain for SOLO.
https://github.com/unity-technologies/pysolotools
billing-5160 owner-computer-vision perception
Last synced: about 1 month ago
JSON representation
Python toolchain for SOLO.
- Host: GitHub
- URL: https://github.com/unity-technologies/pysolotools
- Owner: Unity-Technologies
- License: other
- Created: 2021-11-24T08:56:25.000Z (almost 3 years ago)
- Default Branch: main
- Last Pushed: 2024-07-16T09:13:38.000Z (4 months ago)
- Last Synced: 2024-10-07T09:45:58.688Z (about 1 month ago)
- Topics: billing-5160, owner-computer-vision, perception
- Language: Python
- Homepage: https://Unity-Technologies.github.io/pysolotools
- Size: 304 MB
- Stars: 37
- Watchers: 5
- Forks: 14
- Open Issues: 17
-
Metadata Files:
- Readme: README.md
- Contributing: CONTRIBUTING.md
- License: LICENSE
- Code of conduct: CODE_OF_CONDUCT.md
Awesome Lists containing this project
README
**The open-source tool for loading and analyzing Unity SOLO datasets**
---
[](https://github.com/Unity-Technologies/pysolotools/actions)
[](LICENSE)
## Introduction
pysolotools is a python package for managing SOLO datasets.
It helps to perform following tasks:
- Parse SOLO datasets generated with Unity Perception
- Dataset iterables
- Convert SOLO to [COCO format](https://cocodataset.org/#format-data)
- Compute common statistics
You can read more about SOLO schema [here](https://github.com/Unity-Technologies/perception/blob/main/com.unity.perception/com.unity.perception/Documentation~/SoloSchema/Solo_Schema.md).
## Pre-Requisites
- Install [Anaconda](https://docs.anaconda.com/anaconda/install/) or [Miniconda](https://docs.conda.io/en/latest/miniconda.html) (recommended). Install [pre-commit](https://pre-commit.com/).
- Make sure `pip` is installed.
## Installation
```shell
pip install pysolotools --index-url=https://artifactory.prd.it.unity3d.com/artifactory/api/pypi/pypi/simple
```
#### SOLO Dataset
##### Load and iterate frames
```python
from pysolotools.consumers import Solo
solo = Solo(data_path="src_data_path")
for frame in solo.frames():
# perform operations on frame
```
##### SOLO2COCO conversion
Supports conversion for these labels: 2d bbox, keypoints, instance, semantic.
```python
from pysolotools.converters.solo2coco import SOLO2COCOConverter
from pysolotools.consumers import Solo
solo = Solo("src_data_path")
dataset = SOLO2COCOConverter(solo)
dataset.convert(output_path="output_path")
```
##### Stats computation
Supports bbox, keypoints and image analysis on SOLO dataset.
```python
from pysolotools.consumers import Solo
from pysolotools.stats.analyzers.bbox_analyzer import BBoxHeatMapStatsAnalyzer, BBoxSizeStatsAnalyzer
from pysolotools.stats.handler import StatsHandler
bbheat=BBoxHeatMapStatsAnalyzer()
bbsize=BBoxSizeStatsAnalyzer()
solo = Solo("data_path")
bbh= StatsHandler(solo=solo)
bbh.handle(analyzers=[bbheat,bbsize],cat_ids=[])
```
## Community and Feedback
The Unity Computer Vision demos are open-source and we encourage and welcome contributions.
If you wish to contribute, be sure to review our [contribution guidelines](CONTRIBUTING.md)
and [code of conduct](CODE_OF_CONDUCT.md).
## Support
For feature requests, bugs, or other issues, please file a
[GitHub issue](https://github.com/Unity-Technologies/Unity-Vision-Hub/issues)
using the provided templates we will investigate as soon as possible.
## License
[Apache License 2.0](LICENSE)