Skip to content
Snippets Groups Projects
Commit 0319e4ae authored by dwh48's avatar dwh48
Browse files

Update README.rst

parent b7c2185a
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
------------
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