Skip to content

MMLab-CU/polynet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PolyNet: A Pursuit of Structural Diversity in Very Deep Networks

By Xingcheng Zhang, Zhizhong Li, Chen Change Loy, Dahua Lin

Multimedia Laboratory, The Chinese University of Hong Kong

Homepage pdf

BibTeX

@article{zhang2016polynet,
  title={Polynet: A pursuit of structural diversity in very deep networks},
  author={Zhang, Xingcheng and Li, Zhizhong and Loy, Chen Change and Lin, Dahua},
  journal={arXiv preprint arXiv:1611.05725},
  year={2016}
}

The Very Deep PolyNet

PolyNet

  • Visualization

  • Models

    NOTE: The model is trained using our own deep learning framework Parrots. The caffe model is converted from the Parrots model, and the proto file is based on Yuanjun's fork of Caffe.

Results

model training speed* (#imgs/second) single-crop val top-1 single-crop val top-5 single-crop test top-5 multi-crop val top-1 multi-crop val top-5
ResNet-152 - 22.16 6.16 - 19.38 4.49
ResNet-152^ 279 ( 8 GPUs) 20.93 5.54 5.50 18.50 3.97
ResNet-269^ 245 (16 GPUs) 19.78 4.89 4.82 17.54 3.55
ResNet-500^ 248 (32 GPUs) 19.66 4.78 4.70 17.59 3.63
Inception-v4 - 20.0 5.0 - 1 7.7 3.8
Inception-ResNet-v2 - 19.9 4.9 - 17.8 3.7
Inception-ResNet-v2 314 ( 8 GPUs) 20.05 5.05 5.11 18.41 3.98
Very Deep Inception-ResNet 278 (32 GPUs) 19.10 4.48 4.46 17.39 3.56
Very Deep PolyNet 290 (32 GPUs) 18.71 4.25 4.33 17.36 3.45

^ The ResNet models are trained by Tong Xiao;

* Training speed is measured on Parrots using NVIDIA TITAN X Graphics Cards.

Releases

No releases published

Packages

No packages published