This repo contains the source code for automatic detection
- Linux
- Nvidia drivers, CUDA >= 10.0, cuDNN >= 7
- 32GB of RAM
- 2080ti or another GPU with fp16 support and at least 12GB memory
Download and extract dataset Generate csv
filename,label
./data/1,1
./data/2,0
python train_main.py
If you find our work useful in your research or if you use parts of this code please consider citing our paper:
title = {Sch-net: a deep learning architecture for automatic detection of schizophrenia},
journal = {BioMedical Engineering OnLine},
pages = {1-21},
year = {2021}