fastai выдает ошибку "обучающий набор пуст", когда каталоги не пусты - PullRequest
1 голос
/ 05 мая 2020

Я пытаюсь построить классификатор как с обучающим, так и с тестовым набором. Однако FastAI выдает ошибку UserWarning: Your training set is empty, несмотря на то, что и обучающий, и тестовый наборы не пусты. Это результат моих команд терминала, чтобы узнать, сколько элементов в каждой папке:

%cd Food-101/images/train/
!ls -1 | wc -l

%cd ../test
!ls -1 | wc -l

%cd ../../

/content/Food-101/images/train
75750
/content/Food-101/images/test
25250
/content/Food-101

Это мой код:

import pandas as pd

from fastai import *
from fastai.vision import *
from fastai.callbacks.hooks import *
from pathlib import Path
from numba import vectorize
from subprocess import call, run
import os, git, glob, shutil

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

np.random.seed(42)
data = ImageDataBunch.from_folder(path, train= path + '/images/train', test=path + '/images/test', valid_pct=0.2, ds_tfms=get_transforms(), size=224)
data.normalize(imagenet_stats)

И это ошибка:

/usr/local/lib/python3.6/dist-packages/fastai/data_block.py:458: UserWarning: Your training set is empty. If this is by design, pass `ignore_empty=True` to remove this warning.
  warn("Your training set is empty. If this is by design, pass `ignore_empty=True` to remove this warning.")
/usr/local/lib/python3.6/dist-packages/fastai/data_block.py:461: UserWarning: Your validation set is empty. If this is by design, use `split_none()`
                 or pass `ignore_empty=True` when labelling to remove this warning.
  or pass `ignore_empty=True` when labelling to remove this warning.""")
---------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
/usr/local/lib/python3.6/dist-packages/fastai/data_block.py in get_label_cls(self, labels, label_cls, label_delim, **kwargs)
    264         if label_delim is not None:             return MultiCategoryList
--> 265         try: it = index_row(labels,0)
    266         except: raise Exception("""Can't infer the type of your targets. 

7 frames
IndexError: index 0 is out of bounds for axis 0 with size 0

During handling of the above exception, another exception occurred:

Exception                                 Traceback (most recent call last)
/usr/local/lib/python3.6/dist-packages/fastai/data_block.py in get_label_cls(self, labels, label_cls, label_delim, **kwargs)
    265         try: it = index_row(labels,0)
    266         except: raise Exception("""Can't infer the type of your targets. 
--> 267 It's either because your data source is empty or because your labelling function raised an error.""")
    268         if isinstance(it, (float, np.float32)): return FloatList
    269         if isinstance(try_int(it), (str, Integral)):  return CategoryList

Exception: Can't infer the type of your targets. 
It's either because your data source is empty or because your labelling function raised an error.

Я не понимаю, почему это вызывает эту ошибку, когда оба каталога явно заполнены изображениями.

Ответы [ 2 ]

2 голосов
/ 06 мая 2020

Не могли бы вы убедиться, что ваша переменная path установлена ​​в / content / Food-101 /?

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

Решил проблему:

data = ImageDataBunch.from_folder('/content/Food-101/images', train='/content/Food-101/images/train', test='/content/Food-101/images/test', valid_pct=0.2, ds_tfms=get_transforms(), size=224)
...