From 0319e4ae82bc721c7e43caad7288dc13de99304e Mon Sep 17 00:00:00 2001 From: dwh48 <dwh48@drexel.edu> Date: Thu, 10 Mar 2022 21:27:33 +0000 Subject: [PATCH] Update README.rst --- README.rst | 22 +++++----------------- 1 file changed, 5 insertions(+), 17 deletions(-) diff --git a/README.rst b/README.rst index 38ac21c..09beb1c 100644 --- a/README.rst +++ b/README.rst @@ -1,30 +1,18 @@ -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 -- GitLab