Подгонка модели UNET с набором данных изображений 40k (из ошибки импорта набора данных из load_image набора данных) - PullRequest
0 голосов
/ 29 января 2020

Я новичок в машинном обучении, учусь как я go и работаю над тем, чтобы представить тезисы для конференции, но с трудом подгоняю эту модель к набору данных, с которым я сейчас работаю.

Unet Модель: https://github.com/ternaus/robot-surgery-segmentation/blob/master/Demo.ipynb

Например, когда я использую

from dataset import load_image

Оператор возвращается, заявляя: ModuleNotFoundError:

No module named 'dataset'

Я уже загрузил набор данных, используя Kaggle, но не знаю, почему это происходит, любая помощь будет отличной.

...