Fetching latest commit…
Cannot retrieve the latest commit at this time.

readme.md

name caffemodel caffemodel_url license sha1 caffe_commit
BAIR/BVLC GoogleNet Model
bvlc_googlenet.caffemodel
unrestricted
405fc5acd08a3bb12de8ee5e23a96bec22f08204
bc614d1bd91896e3faceaf40b23b72dab47d44f5

This model is a replication of the model described in the GoogleNet publication. We would like to thank Christian Szegedy for all his help in the replication of GoogleNet model.

Differences:

  • not training with the relighting data-augmentation;
  • not training with the scale or aspect-ratio data-augmentation;
  • uses "xavier" to initialize the weights instead of "gaussian";
  • quick_solver.prototxt uses a different learning rate decay policy than the original solver.prototxt, that allows a much faster training (60 epochs vs 250 epochs);

The bundled model is the iteration 2,400,000 snapshot (60 epochs) using quick_solver.prototxt

This bundled model obtains a top-1 accuracy 68.7% (31.3% error) and a top-5 accuracy 88.9% (11.1% error) on the validation set, using just the center crop. (Using the average of 10 crops, (4 + 1 center) * 2 mirror, should obtain a bit higher accuracy.)

Timings for bvlc_googlenet with cuDNN using batch_size:128 on a K40c:

  • Average Forward pass: 562.841 ms.
  • Average Backward pass: 1123.84 ms.
  • Average Forward-Backward: 1688.8 ms.

This model was trained by Sergio Guadarrama @sguada

License

This model is released for unrestricted use.