ImageDataBunch.from_folder не видит мой тестовый набор - PullRequest
Новая
       83

ImageDataBunch.from_folder не видит мой тестовый набор

0 голосов
/ 03 мая 2020

Я работаю над набором данных Food-101 и пытаюсь предоставить ImageDataBunch.from_folder () с отдельным набором тестов, но по какой-то причине он не распознает его. 

ImageDataBunch;

Train: LabelList (60600 items)
x: ImageList
Image (3, 224, 224),Image (3, 224, 224),Image (3, 224, 224),Image (3, 224, 224),Image (3, 224, 224)
y: CategoryList
train,train,train,train,train
Path: images/train;

Valid: LabelList (15150 items)
x: ImageList
Image (3, 224, 224),Image (3, 224, 224),Image (3, 224, 224),Image (3, 224, 224),Image (3, 224, 224)
y: CategoryList
train,train,train,train,train
Path: images/train;

Test: None

Когда у меня была эта проблема ранее, мне сказали, что я должен держать свой поезд и тестировать изображения на одном уровне. Это моя файловая структура atm: 

Food-101
      images
            train
               ...75750 train images
            test
               ...25250 test images
      meta
            classes.txt
            labes.txt
            test.json
            test.txt
            train.json
            train.txt
      README.txt
      license_agreement.txt

И это в основном код, который я использую: 

train_image_path = Path('images/train/')
test_image_path = Path('images/test/')
path = Path('../Food-101')

food_names = get_image_files(train_image_path)

file_parse = r'/([^/]+)_\d+\.(png|jpg|jpeg)$'

data = ImageDataBunch.from_folder(train_image_path, test_image_path, valid_pct=0.2, ds_tfms=get_transforms(), size=224)
data.normalize(imagenet_stats)
...