Skip to content

annedranowski/knot-cnn

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

89 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

🪢 Knot Recognition with CNN

This repository contains a lightweight convolutional neural network (CNN) for detecting knots in images.
It includes:

  • A training pipeline (src/train.py)
  • Dataset utilities (src/dataset.py)
  • A model definition (src/models.py)
  • A clean demo notebook (notebooks/knot_detector.ipynb)

The project is structured for easy local training, evaluation, and future deployment on Hugging Face Spaces.


🚀 Quickstart

1. Clone the repository

git clone https://github.com/annedranowski/knot-cnn.git
cd knot-cnn

HF integration

Script to snapshot the HF dataset in data/raw/....

Run

python scripts/sync_hf_data.py \
  --repo-id tr33hugg3r/knot-crossings \
  --revision main

About

Build and train a knot-detector CNN

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 4

  •  
  •  
  •  
  •