Train the model with DIGITS

weiliu89 / caffe

Caffe: a fast open framework for deep learning.

http://caffe.berkeleyvision.org/

Other

4.77k stars 1.67k forks source link

Train the model with DIGITS #394

Open MyVanitar opened 7 years ago

MyVanitar commented 7 years ago

is it possible to train SSD with Nvidia DIGITS with my own dataset (images)?

weiliu89 commented 7 years ago

I never used DIGITS before, and I don't know. Maybe you could merge the ssd branch with DIGITS branch and resolve the conflicts, but I am not sure.

MyVanitar commented 7 years ago

Actually if your algorithm and Model is based on Caffe, then DIGITS can handle it, but the DIGITS does only work with annotations files in KITTI format. DIGITS is actually a GUI for command line operations of the Caffe.

Anyway, if I decided to use a pre-trained model and adjust it to detect custom class(es), actually fine-tune the model, then which places in the prototxt file should be changed?