bomri / code-for-posts Goto Github PK

Thank you for your posts
I have a problem about imagefolder in pytorch?

AttributeError Traceback (most recent call last)
in
17 sampler=BalancedBatchSchedulerSampler(dataset=image_datasets['train'],batch_size=16),
18 shuffle=False, num_workers=4)
---> 19 for x in ['train', 'val', 'test']}
20
21 dataset_sizes = {x: len(image_datasets[x]) for x in ['train', 'val', 'test']}

in (.0)
17 sampler=BalancedBatchSchedulerSampler(dataset=image_datasets['train'],batch_size=16),
18 shuffle=False, num_workers=4)
---> 19 for x in ['train', 'val', 'test']}
20
21 dataset_sizes = {x: len(image_datasets[x]) for x in ['train', 'val', 'test']}

~/ml/pytorch-image-classification/imbalancedbatch/balanced_sampler.py in init(self, dataset, batch_size)
22 self.dataset = dataset
23 self.batch_size = batch_size
---> 24 self.number_of_datasets = len(dataset.datasets)
25
26 def len(self):

AttributeError: 'ImageFolder' object has no attribute 'datasets'