Skip to content
Snippets Groups Projects
Commit 5276af6d authored by hannandarryl's avatar hannandarryl
Browse files

Merge remote-tracking branch 'origin/main' into main

parents c1d031a1 0319e4ae
Branches
No related tags found
No related merge requests found
Sparse Coding Library for PyTorch Sparse Coding Library for Tensorflow
================================= =================================
Installation
------------
To install the package, clone it to a local folder, then run `pip install -e .` from the folder to install the package in edit mode (so changes to the code are used).
Overview Overview
------------ ------------
Main sparse coding file is found at: Main sparse coding file is found at:
scripts/train_conv3d_sparse_model.py keras/train_sparse_model.py
Sparse coding command example: Sparse coding command example:
python scripts/train_conv3d_sparse_model.py python train_sparse_model.py
Main classification model is found at: Main classification model is found at:
scripts/train_classifier.py keras/train_classifier.py
Classification command example: Classification command example:
python data_classifiers/small_data_classifier.py --checkpoint sparse.pt python small_data_classifier.py --checkpoint sparse.pt
Run on a single Yolo clip with:
python scripts/run_on_single_image.py
Notes:
Need to have classifier.pt and sparse.pt (checkpoints) in root
Videos go in a directory called input_videos
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment