почему ToPILImage () может получить лучший результат MESloss? - PullRequest
Новая
       51

почему ToPILImage () может получить лучший результат MESloss?

0 голосов
/ 30 апреля 2020

Применение CNN к фильтру изображения, ввод в формате YUV444. Если не применять ToPILImage (), данные поезда - это просто тензор факела из массива numpy, MSELoss составляет около 10k. Но, если применить ToPILImage (), данные поезда преобразуются следующим образом: 

  def transform(self, x):
      train_transform = transforms.Compose([
      transforms.ToPILImage(mode='YCbCr'),
      transforms.ToTensor(),])
      return train_transform(x)

Значение MSELoss ниже 10. После Google, похоже, нет объяснения этому. Будучи новичком в pytorch и NN, я действительно смущен этим.

...