Imbalanced Dataset Sampler是一个针对不均衡数据集的重采样器。但直接使用自定义数据集时会报错。
查GitHub发现Imbalanced Dataset Sampler需要调用数据集的get_labels() 用于返回你数据集所有的标签值,而自定义的数据集一般是没有这个方法的。
所以在自己的数据集类里定义一个输出自己全部标签的方法就可以了,同时使用Sampler就不能在dataloader中使用shuffle了
Imbalanced Dataset Sampler是一个针对不均衡数据集的重采样器。但直接使用自定义数据集时会报错。
查GitHub发现Imbalanced Dataset Sampler需要调用数据集的get_labels() 用于返回你数据集所有的标签值,而自定义的数据集一般是没有这个方法的。
所以在自己的数据集类里定义一个输出自己全部标签的方法就可以了,同时使用Sampler就不能在dataloader中使用shuffle了