liga-pytorch
: PyTorch Model Flavor on Ligaliga-tv
: PyTorch Model Flavor and Torch Vision Model Types on Liga
pip install liga-tv
With liga-tv
, one can analyze video via SQL and pre-trained PyTorch Vision models in a Liga way.
For ML Engineers who train their own PyTorch models and log the models into MLflow, liga-mlflow
is suggested.
Try the live notebooks via:
# Please install JDK 8/11, Python 3.8/3.9 locally
git clone https://github.com/liga-ai/liga-pytorch.git
cd liga-pytorch && bin/lab
Model Type (Short) | Model Type (Long) | Purpose | Preview Latest Notebook | Try Google Colab Notebook |
---|---|---|---|---|
convnext | convnext_base |
Classification | Demo | |
convnext_tiny |
||||
convnext_small |
||||
convnext_large |
||||
efficientnet | efficientnet_b0 |
Classification | Demo | |
efficientnet_b[1-7] |
||||
fasterrcnn | fasterrcnn |
Object Detection | Demo | |
fasterrcnn_mobilenet_v3_large_320_fpn |
||||
fasterrcnn_mobilenet_v3_large_fpn |
||||
fasterrcnn_resnet50_fpn |
||||
keypointrcnn | keypointrcnn |
Object Detection | Demo | |
maskrcnn | maskrcnn |
Object Detection | Demo | |
resnet | resnet50 |
Classification | Demo | |
resnet18 |
||||
resnet34 |
||||
resnet101 |
||||
resnet152 |
||||
ssd | ssd |
Object Detection | Demo | |
ssdlite |