NVIDIA / caffe

Caffe: a fast open framework for deep learning.
http://caffe.berkeleyvision.org/
Other
672 stars 263 forks source link

Something about "Packing packing = NHWC" in annotated_data_layer #580

Open chenpengf0223 opened 4 years ago

chenpengf0223 commented 4 years ago

Hi, Thanks for your great work. I try to train a wider face detection model. But I meet some errors as following: image

I feel that annotated_data_layer.cpp,line182: Packing packing = NHWC; should be changed to : Packing packing = NCHW. Could you please give me some guidance? Thanks again.

92ypli commented 4 years ago

have you solver this problem? how to do ?

chenpengf0223 commented 4 years ago

have you solver this problem? how to do ?

@92ypli

set it to 'NCHW': Packing packing = NCHW.