Supported Modules#
The table below shows all the current supported modules which are integrated in pyppbox in many different ways. The main idea of the integrations is to make them fully compatible with our PoseTReID framework read more here. Thus, not every original feature of every supported module is fully functioning as it was made. For example, the supported YOLO_Classic / YOLO module in the table can only be used as a detector (Inference/Prediction) using the native or official `.weights` files, but it canβt do any custom data traning, etc.
Modules |
General Name |
Config Name |
Details |
---|---|---|---|
Detectors |
YOLO |
YOLO_Classic |
|
YOLO_Ultralytics |
* Integrated by linking ultralytics
* Model: .pt V3, V5, V8, V9, v10, v11
* Run on: CPU or GPU (PyTorch)
|
||
Trackers |
Centroid |
Centroid |
* Built-in / Native
* Run on: CPU
|
SORT |
SORT |
||
DeepSORT |
DeepSORT |
||
ReIDers |
FaceNet |
FaceNet |
|
Torchreid |
Torchreid |
* Integrated by linking pyppbox-torchreid
* Model: OSNet-AIN, OSNet, MLFN
* Run on: CPU or GPU (PyTorch)
|