1. ValueError: With n_samples=0, test_size=0.2 and train_size=None, the resulting train set will be empty. Adjust any of the aforementioned parameters.
解决方案:
修改图片路径,改成绝对路径
2.AttributeError: module 'albumentations.augmentations.transforms' has no attribute 'RandomRotate90'
导入一个新的包
依次修改,将原来的注释掉,改成下图中代码,由多个类似问题,依次修改即可
代码如下:
albu.RandomRotate90(),
albu.Flip(),
lbu.Resize(config['input_h'], config['input_w']),