A multi-purpose Video Labeling GUI in Python with integrated SOTA detector and tracker.
Project description
UltimateLabeling
A multi-purpose Video Labeling GUI in Python with integrated SOTA detector and tracker. Developed using PyQt5.
[Under development]
The integrated object detectors and trackers are based on the following codes:
- OpenPifPaf: for human pose estimation
- YOLO darknet: for object detection
- SiamMask: for visual object tracking
- Hungarian algorithm (scipy.optimize): for optimal instance ID assignment
For remote server processing, follow the guide below in order to configure the server files.
Demo
Features
- SSH connection to the remote server (see below to configure the server)
- YOLO and OpenPifPaf integrated object & pose detectors (single frame and entire video mode)
- Hungarian instance ID assignment
- SiamMask Visual object tracking for missing or mislabeled objects
- Zoom on the video, resizable bounding boxes and skeletons
- Dark mode!
Installation
Install the requirements using pip
as follows:
pip install -r requirements
Put the videos (folder of images or video file, the frames will be extracted automatically) inside the data
folder. Then, open the GUI using
python -m ultimatelabeling.main
After closing the window, the annotations are available in the output
folder.
Remote server configuration
To configure the remote GPU server, follow the steps below:
git clone https://github.com/alexandre01/UltimateLabeling_server.git
cd UltimateLabeling_server
pip install -r requirements.txt
bash siamMask/setup.sh
bash detection/setup.sh
The data images and videos should be placed in the folder data
, similarly to the client code.
To extract video files, use the following script:
bash extract.sh data/video_file.mp4
Licence
Copyright (c) 2019 Alexandre Carlier, released under the MIT licence.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
File details
Details for the file ultimatelabeling-0.1.1.tar.gz
.
File metadata
- Download URL: ultimatelabeling-0.1.1.tar.gz
- Upload date:
- Size: 38.7 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/1.13.0 pkginfo/1.5.0.1 requests/2.22.0 setuptools/41.0.1 requests-toolbelt/0.9.1 tqdm/4.32.1 CPython/3.7.3
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 |
6085dd694d329bb2f26ee6e23b01f867bab8598f9417949bc4e588b0ab3f5e1a
|
|
MD5 |
c3c93689077978899c8df15def6e06da
|
|
BLAKE2b-256 |
72cfeeb6bc2c4d76deda9b130a53dbc6be8e27f86523ead665d9855576665b20
|