GOTURN icon indicating copy to clipboard operation
GOTURN copied to clipboard

A question about pre-train model?

Open Ouya-Bytes opened this issue 8 years ago • 15 comments

Dear davheld: I have a question about how do you get the pretrain model, can you give some detials? Question as follows: (1) Which dataset ( such:ILSVRC2014_DET or ILSVRC2014_CLS) to pretrain the convolutional layer? (2) If I only use the dataset of ALOV300+ and 2014_DET to train the net get regression value , or don't do the step of pretrain convolution layer, will decresed the tracking performance? (3) When you train the siamese net, the two network branch will shared the layer params? or has independence parameters? Looking forward to your reply, best wish!

Ouya-Bytes avatar Mar 04 '17 12:03 Ouya-Bytes

  1. I use the pre-trained CaffeNet architecture, which is available from Caffe: http://caffe.berkeleyvision.org/model_zoo.html

  2. I think it probably will - if you do this then I recommend using a much smaller architecture.

  3. As mentioned above, I do not pre-train the convolutional layers myself but I take the layers pre-trained from Caffe. The two sets of convolutional layers have identical weights.

davheld avatar Mar 05 '17 01:03 davheld

@davheld I according to your code(tracker.prototxt and solver.prototxt ,iteration 500000) and use the train.cpp to train the network use dataset(2014_DET and ALOV300+), the train loss value is not convergent, and oscillation between on 20 ~ 50 finally.so that have a very pool tracking performance. Can you give me some advise? Thks

Ouya-Bytes avatar Mar 05 '17 13:03 Ouya-Bytes

It sounds like you are overfitting. Just to be sure - I don't train the conv layers at all, those are pre-trained using CaffeNet.

davheld avatar Mar 05 '17 14:03 davheld

yet, i only use your code and prototxt( run the train.cpp, keep params of solver.prototxt and tracker.prototxt) to re-train the network, i don't change anymore. the convolutional layer is from CaffeNet, and lr_mult is set 0 no change.

Ouya-Bytes avatar Mar 05 '17 14:03 Ouya-Bytes

How do you create the pre-trained network?

davheld avatar Mar 05 '17 15:03 davheld

the pretrian param from your offer address http://cs.stanford.edu/people/davheld/public/GOTURN/weights_init/tracker_init.caffemodel, i dont change the prototxt, i only want to run train.cpp code to get tracker_iter_500000.caffemodel, then can test the tracker

Ouya-Bytes avatar Mar 05 '17 15:03 Ouya-Bytes

That's odd, not sure.

davheld avatar Mar 05 '17 15:03 davheld

I have the same problem, changing val_ratio from 0.2 to 0 in "loader/loader_alov.cpp" may help, but still , model trained by myself doesn't perform as good as pre-train model.

Jiangfeng-Xiong avatar Mar 06 '17 02:03 Jiangfeng-Xiong

not convergent? oscillation?

Ouya-Bytes avatar Mar 06 '17 02:03 Ouya-Bytes

The oscillation is normal and simply occurs because the training evaluation is occurring on mini-batches which are randomly sampled at each iteration. However, the numbers that you listed seem lower than what I remember so I believe that you are overfitting, although I am not sure why.

davheld avatar Mar 06 '17 02:03 davheld

train loss is like this train loss, range from 20 to 90 @OuYag

Jiangfeng-Xiong avatar Mar 06 '17 02:03 Jiangfeng-Xiong

Try reducing the learning rate? The oscillations in that graph look fairly large. Although if you are using the default learning rate then this is unusual that you should need to change it. Also, make sure that all convolutional layers are fixed (e.g. in both streams of the network).

davheld avatar Mar 06 '17 02:03 davheld

@OuYag your words"lr_mult is set 0 no change." I don't think it is right, lr_mult set to 0 means no learning rate. Caffe says that we will set the weight learning rate to be the same as the learning rate given by the solver during runtime

ujsyehao avatar Mar 15 '17 06:03 ujsyehao

@Jiangfeng-Xiong @OuYag do you solve the issue? I have the same problem, the test loss value is between 10 and 20. I guess it is overfitting, however changing lr or batchsize cannot reduce losses

freescar avatar Nov 01 '17 04:11 freescar

Hi, I want to know how to evaluate your performance?

wendianwei avatar Nov 03 '17 07:11 wendianwei