How we do transfer learning using pretrained models with grey scale images as input

조회 수: 9 (최근 30일)
if we planned to do a training using greyscale images with pretrained model. It will show error. Is there is any method to train these type of images. That means apart from RGB images, how we do the transfer learning

채택된 답변

Abhishek Singh
Abhishek Singh 2019년 3월 8일
There might be many ways to go about it:
1. Copy and stack the image matrix three times to replicate the 3 channels of RGB
2. Stack two zero or identity matrices on top of the grey image matrix such that the Red channel has all the data and the other two channels are just place holders (not advisable)
3. Remove the other channel weights in the pre-trained model itself (very cumbersome)
This is an open-ended problem with many other creative solutions like using DeOldify GAN to color them first or just interpolating the other two channels in some way. The way to find the best solution would be to experiment with your data and pre-trained architecture

추가 답변 (0개)

카테고리

Help CenterFile Exchange에서 Deep Learning Toolbox에 대해 자세히 알아보기

제품


릴리스

R2018b

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by