Skip to content

ChristianHooper/Scratch-CNN

Repository files navigation

Scratch-CNN

A modular CNN for detection and classification build with NumPy & Cython from scratch.

Setup

Create an environment with Conda:

conda env create --file standard_environment.yml --name scratch
conda activate scratch

Run

Execute the sample pipeline on the test data:

python src/cnn_base.py

Optional Cython acceleration

Build the fast_ops extension with:

python -m pip install -e .

If the extension is not built, the code falls back to the Python implementation. Right now only the max pooling function is supported.

About

A modular CNN for detection and classification build with NumPy from scratch.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors