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

Acheiving Comparable performance using Resnet #4

Open
varghesealex90 opened this issue Feb 21, 2018 · 16 comments
Open

Acheiving Comparable performance using Resnet #4

varghesealex90 opened this issue Feb 21, 2018 · 16 comments

Comments

@varghesealex90
Copy link

Hi Zoozog;

This is not an issue, rather an update.
Using your code as the base, I trained a ResNet-18 ( PRETRAINED on imagenet) for the CXR task. I see that the performance is comparable to a large extent. I assume that ResNET18 would have fewer parameters than densenet121.

AUROC mean 0.847266414478356
Atelectasis 0.8323852409157189
Cardiomegaly 0.9118541996631041
Effusion 0.8866060621559915
Infiltration 0.7162746001795058
Mass 0.8627326595643292
Nodule 0.7897284404542668
Pneumonia 0.7753048198007456
Pneumothorax 0.8821548678140269
Consolidation 0.8131018114945643
Edema 0.8982024375815074
Emphysema 0.9355988306093312
Fibrosis 0.8532190562301323
Pleural_Thickening 0.7942477917778453
Hernia 0.9103189844559144

I am also experimenting with training the network from scratch ( sans pre-trained weights). Shall update you once that is done.

@zoogzog
Copy link
Owner

zoogzog commented Feb 22, 2018

Nice results! Thanks for sharing!

@Ahsanr312
Copy link

Hi varghesealex90;
Can you share your code for ResNet-18 ?

@varghesealex90
Copy link
Author

varghesealex90 commented Jul 14, 2019 via email

@Ahsanr312
Copy link

Ahsanr312 commented Jul 14, 2019 via email

@varghesealex90
Copy link
Author

varghesealex90 commented Jul 14, 2019 via email

@Ahsanr312
Copy link

Ahsanr312 commented Jul 17, 2019 via email

@Ahsanr312
Copy link

Ahsanr312 commented Jul 17, 2019 via email

@varghesealex90
Copy link
Author

varghesealex90 commented Jul 18, 2019 via email

@Ahsanr312
Copy link

Ahsanr312 commented Jul 18, 2019 via email

@Ahsanr312
Copy link

Ahsanr312 commented Jul 19, 2019 via email

@prasanth-rgb
Copy link

Hi,
My question may be stupid. Anyway, i am also in a struggle for building a mode for Chest XRay classification.

I have read that for transfer learning we have to find a model which had already trained with similar data set (medical image of chest xray). But here it is using Imagenet right?

Can we use Imagenet trained model for retraining with chexpert data set?

@varghesealex90
Copy link
Author

varghesealex90 commented Dec 5, 2019 via email

@Aliktk
Copy link

Aliktk commented Jun 1, 2021

@varghesealex90 it's too late but as a researcher, I found this study and going to replicate it. Now is it possible now to help in replication?
If yes then I am sharing what I have done.
I want to retrain the model by changing the runTrain() and putting data into their folder as required.
I use the debugger to run the code as many errors coming during. The model loaded for training but before starting training it gives error which I mentioned after your response.
What was your system specs, what packages version you installed.
I am using Window IBM inctance with 64GB Ram and 16GB GPU.
Need you help thanks

@varghesealex90
Copy link
Author

Hi @Aliktk can you please paste your errors. I ran the code on Ubuntu Machine and probably on a GPU with 12 Gb

@Aliktk
Copy link

Aliktk commented Jun 1, 2021

@varghesealex90 Thank you so much for your quick response:
I have installed
python 3.8.0
CUDA.11
torch1.8.1
torchaudio 0.8.1
tensorflow-gpu 2.4.1
opencv-python 4.5.1.48
If I installed the wrong version please correct me. Also, tell me a simple method to replicate with same results
Following are the few problems that I have to suffer till.
Error-1: When I run the script it says you have an old version of CUDA please update the CUDA by this this....(RuntimeError: The NVIDIA driver on your system is too old (found version 10020). Please update your GPU driver by downloading and installing a new version from)
Error-2: AttributeError: Can't pickle local object 'ChexnetTrainer.test..'
Error-3: self = reduction.pickle.load(from_parent) EOFError: Ran out of input
Please tell me the exact version which I installed to relicate

@varghesealex90
Copy link
Author

As mentioned previously, I did this activity 3 years back. I am guessing pytorch version should be 0.4.0

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

5 participants