caffe-cvprw15 icon indicating copy to clipboard operation
caffe-cvprw15 copied to clipboard

How I can train myself datasets with ImageNe pretrained model ?

Open aiaini66 opened this issue 7 years ago • 4 comments

I have read the README.md file and see your words:"You need to convert the dataset into leveldb format using "create_imagenet.sh". We will show you how to do this. To be continued.". Now I want to train myself Datasets with ImageNe pretrained model ,if there are any instruction about how to convert myself Datasets format? I'm forward to your reply!

aiaini66 avatar Jan 18 '18 05:01 aiaini66

  1. Change the paths in create_imagenet.sh with your dataset path. $ vi /examples/cvprw15-cifar10/create_imagenet.sh

  2. Run script prepare_train.sh to download ImageNet pretrained model and convert myself dataset to leveldb format. $ ./prepare_train.sh

  3. Change the source with your leveldb source in training script. $ vi /examples/cvprw15-cifar10/train_48.sh

  4. Go to the folder /examples/cvprw15-cifar10, and run the training script:

$ cd /examples/cvprw15-cifar10 $ chmod 777 train_48.sh
$ ./train_48.sh

  1. You will finally get your model named KevinNet_CIFAR10_48_iter_xxxxxx.caffemodel. To use your model, modify the model_file in demo.m to link to your model:

    model_file = './YOUR/MODEL/PATH/filename.caffemodel';

  2. Launch matlab and run demo.m.

brijesh099 avatar Jan 18 '18 05:01 brijesh099

thanks a lot! If I have any questions in the further work ,I'll ask you for help ! best wish!

aiaini66 avatar Jan 19 '18 02:01 aiaini66

The baidu link of model and image resource is invalid, can anyone share it?

mudi-1125 avatar Mar 08 '18 04:03 mudi-1125

About the baidu link, I found that my baidu account is dead since Baidu request new authentication. It only accept China's phone number, but I don't have one. Sorry for the convenience.

kevinlin311tw avatar Apr 17 '18 08:04 kevinlin311tw