search

coxlab / prednet

Code and models accompanying "Deep Predictive Coding Networks for Video Prediction and Unsupervised Learning"
https://arxiv.org/abs/1605.08104
MIT License
759 stars 259 forks source link

Non-RGB Images #52

Closed mitkina closed 5 years ago

mitkina commented 5 years ago

Hello,

Is there anything that needs to be changed to pass in images with channel size of 1 instead of 3 as in RGB within the model (besides setting n_channels)?

Thanks!

matinhosseiny commented 5 years ago

just change this line : n_channels, im_height, im_width = (1, 64, 64) in kitti_train.py

mitkina commented 5 years ago

Thanks! I realized I was having batch size/learning rate tuning issues!