Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

train on CQ500 #4

Open
AI678 opened this issue Apr 16, 2021 · 2 comments
Open

train on CQ500 #4

AI678 opened this issue Apr 16, 2021 · 2 comments

Comments

@AI678
Copy link

AI678 commented Apr 16, 2021

Thanks for sharing your great work. The paper claimed that cnn-lstm model was trained on Kaggle dataset and further tested on the CQ500 dataset.
Can the code provided in this repository be used to train a cnn-lstm model using train data from CQ500 ?

@nhannguyen2709
Copy link
Collaborator

Hi @AI678,
You can train with any brain-ct classification dataset following the same data-preprocessing + modeling pipeline from this repo e.g. 3-windowed RGB-like images + CNN-LSTM architecture.
CQ500 was originally created by the Qure.ai team (http://headctstudy.qure.ai/), which can be downloaded, contained only 500 studies. The larger training set (Qure25k) is not publicly available.

500 studies is a small number so I advise you against training models solely on the CQ500 dataset.

@sureshsivakumar
Copy link

@nhannguyen2709 can you please share the Hardware requirement and the package requirements to execute this project for learning purpose ?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants