search

zengarden / light_head_rcnn

Light-Head R-CNN
833 stars 223 forks source link

some suggestion about to accelerate speed #27

Open machanic opened 6 years ago

machanic commented 6 years ago

some suggestion about to accelerate speed I found your data format is "NHWC" , but according to page : https://www.tensorflow.org/performance/performance_guide NHWC is the TensorFlow default and NCHW is the optimal format to use when training on NVIDIA GPUs using cuDNN. You should use NCHW as your CNN format, and your nms, ps roi pooling may also need change.

machanic commented 6 years ago

If you use GPU, you should always use NCHW