search

emilwallner / Coloring-greyscale-images

Coloring black and white images with deep learning.
MIT License
1.03k stars 227 forks source link

Inception output size #6

Closed Cwei1 closed 5 years ago

Cwei1 commented 5 years ago

Hi! I'm just curious as to why your model uses a shape 1000 long instead of 1001, as in the paper?

emilwallner commented 5 years ago

It uses the classes from the pre-trained model provided by Keras. Some pre-trained model vary slightly.