gustavla / autocolorize

Automatic colorizaton of grayscale images using Deep Learning
BSD 3-Clause "New" or "Revised" License
220 stars 65 forks source link

difference in autocolorize.prototxt and train_vgg16.prototxt #12

Open yt605155624 opened 6 years ago

yt605155624 commented 6 years ago

Hi, I have view the structure of autocolorize.prototxt and train_vgg16.prototxt in netscope,I think the train_vgg16.prototxt is for training and the autocolorize.prototxt is for inference,but I found that except the layers in VGG, there are many layers in autocolorize.prototxt but not in train_vgg16.prototxt ,for example, _reshaped,_full_reshaped,*_full,prediction_h,prediction_c, many of these layers have params to learn when there type is conv or deconv,but there are no these layers in the training protxt,so How can you get these params when you inference? Could you please help ,me,Thank you!