augmented-neural-odes icon indicating copy to clipboard operation
augmented-neural-odes copied to clipboard

Not making it to run the code for images (Request!)

Open asifehmad opened this issue 4 years ago • 6 comments

Hello! Thank you so much for the work and providing codes. I am new to this field. I tried the code on Google Colab following the pattern you have provided, but my efforts went in vain (Mainly, the dataloader.). I would be very thankful if you provide any notebook for any of the image dataset. Regards

asifehmad avatar Nov 23 '19 09:11 asifehmad

Or any guidance in this regard would be much appreciated, sir.

asifehmad avatar Nov 23 '19 16:11 asifehmad

Hi! To run code on image datasets I would recommend looking at the experiments_img.py file. This contains code to run experiments on image datasets. It should be pretty easy to take this code and port it to a notebook 😄 Hope that helps!

EmilienDupont avatar Nov 27 '19 21:11 EmilienDupont

Hi! To run code on image datasets I would recommend looking at the experiments_img.py file. This contains code to run experiments on image datasets. It should be pretty easy to take this code and port it to a notebook 😄 Hope that helps!

Hi! Thank you so much for the response, sir!

asifehmad avatar Dec 08 '19 05:12 asifehmad

Hi, sir! I want to test on brain MRI data set. What would you suggest in this regard? Would be very much thankful!

asifehmad avatar Dec 09 '19 07:12 asifehmad

Hi, if you want to test on brain MRI data, I would suggest trying to build a dataloader for this dataset that looks exactly like the ones for MNIST or CIFAR10, i.e. each dataset item should return a tuple of an image (as a tensor) and a label (as an int). If you put this dataloader in the dataloaders.py file you can then import it at the top of experiments_img.py. You should then simply add a few extra lines here with something like

if dataset == 'brain_mri':
  data_loader, test_loader = brain_mri_dataloader(training_config["batch_size"])
  img_size = (1, 32, 32)  # Put your image sizes here
  output_dim = 10  # Put your number of classes here

Good luck!

EmilienDupont avatar Dec 21 '19 20:12 EmilienDupont

Hi, if you want to test on brain MRI data, I would suggest trying to build a dataloader for this dataset that looks exactly like the ones for MNIST or CIFAR10, i.e. each dataset item should return a tuple of an image (as a tensor) and a label (as an int). If you put this dataloader in the dataloaders.py file you can then import it at the top of experiments_img.py. You should then simply add a few extra lines here with something like

if dataset == 'brain_mri':
  data_loader, test_loader = brain_mri_dataloader(training_config["batch_size"])
  img_size = (1, 32, 32)  # Put your image sizes here
  output_dim = 10  # Put your number of classes here

Good luck!

Thank you so much, sir!

asifehmad avatar Dec 22 '19 07:12 asifehmad