Оси не совпадают с ошибкой массива в pytorch

Я пытался запустить реализацию vnet в pytorch ( https://github.com/mattmacy/vnet.pytorch) и после нормализации сканирования с

x_max = 512 
y_max = 512 
z_max = 500 
voxspacing = 0.7

когда я звоню tran Функция на линии, где цикл для перечисления через загрузчик данных я получаю

оси не совпадают с ошибкой массива.

Я не понимаю, как это исправить. Любая помощь будет очень полезна. Я новичок в машинном обучении

Источник

1 ответ

Решение

Прокомментируйте строки 417 и 418. проблема будет исправлена

проблема из-за этих 2 строк

if self.transform is not None:
     img = self.transform(img)
Источник
Другие вопросы по тегам