падает в эксепшн 'NoneType' object has no attribute 'to', потому что mtcnn(img) возвращает объект <class 'NoneType'>
можно как-то в ней проверить тип mtcnn(img) перед вызовом to?
что-то типа ?.
if mtcnn(img): mtcnn(img).to(device)
через лямбду можно (lambda x: x.to(device) if x else x)(mtcnn(img))
Два раза вызывать зачем
вот да, я щас размышляю, и это какой-то не верный путь. я хочу на gpu ее считать, а проверка при этом на cpu будет происходить...
Что-то вспомнил опять Ваш кусок. Здесь же все банально просто res = mtcnn(img) if res is not None: res.to(device) И обработка исключения не нужна
Обсуждают сегодня