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
No related branches found
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
------------
Main sparse coding file is found at:
scripts/train_conv3d_sparse_model.py
keras/train_sparse_model.py
Sparse coding command example:
python scripts/train_conv3d_sparse_model.py
python train_sparse_model.py
Main classification model is found at:
scripts/train_classifier.py
keras/train_classifier.py
Classification command example:
python data_classifiers/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
python small_data_classifier.py --checkpoint sparse.pt
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment