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