search

rezoo / illustration2vec

A simple deep learning library for estimating a set of tags and extracting semantic feature vectors from given illustrations.
MIT License
669 stars 112 forks source link

Train network #3

Open bobobo1618 opened 8 years ago

bobobo1618 commented 8 years ago

A basic view of how the network was originally trained would be very helpful. I have a larger dataset than was originally used and I'm hoping to evaluate your network and compare the results to Google's Inception network.

Without knowing what kinds of preprocessing were used (this can be partially inferred from the code provided) and how exactly the network was trained, this is difficult though.

Adding the solver prototxt that was used to train may be enough.

rezoo commented 8 years ago

Sorry for replying late.

Thank you for your suggestion. Providing protobuf files used for training is a good idea. I'm now searching these files...

AlexCSDev commented 7 years ago

Hey @rezoo, have you found source files for the network? I wanted to try to train the network on my own data like bobobo1618 but I'm complete newbie in neural networks and I'm not sure how to train it using only 2 network configuration files which you have provided on the website.