search

rpng / calc

Convolutional Autoencoder for Loop Closure
BSD 3-Clause "New" or "Revised" License
190 stars 45 forks source link

Question about training on my own dataset #6

Closed zqnnn closed 6 years ago

zqnnn commented 6 years ago

Hello, if I want to use your net model train on my dataset, what should I do?

nmerrill67 commented 6 years ago

Hello,

If I am understanding correctly, you want to fine tune our pretrained model with your own dataset?

If that's the case, then first create the LMDBs for your dataset. You can see how to do this by running ./main.py db -h. Make sure you don't specify the --train-after flag. After those are made, you can run ./main.py net --define --train -w model/calc.caffemodel -x1 <your X1> -x2 <your X2> to define and train the model, initializing the weights to the previous model. Simply skip the -w model/calc.caffemodel to train from scratch. You can change the learning rate in makeNet.py in the line self.sp['base_lr'] = '0.0009' #'0.0018' if you want to lower it for fine tuning.

zqnnn commented 6 years ago

Thank you. I have read your paper. I don't know why you say that "Without any extra optimization constraints, the network learned zero vectors". I want to reconstruct the raw image. 2018-07-12 09 21 20

nmerrill67 commented 6 years ago

Right. I think we should take this conversation off the issues feed. Please email me at nmerrill@udel.edu with questions about the paper specifically.

nmerrill67 commented 6 years ago

Closing for now since this is not specifically related to the repo any more.