Efficiently-Updatable Neural Network (NNUE) Refactor of Classic U-Net Architecture for Membrane Segmentation

Developed by Luzhou Zhang - Project still under development

🧠

Setup

Clone into repository: git clone https://github.com/ThaumielSparrow/cremi-nnue

Install dependencies: pip install -r requirements.txt

Download CREMI training and test data here.

Modify runtime variables in main.py and train.py and run program: python main.py

Note: This project has only been tested and validated for Python 3.9.X and 3.10.X with frozen packages. It is likely that any Python version >3.7 supports it.

Docs

I'm not writing documentation lol