cpm_data_transformer corrupted

[wujiyoung]

Hi, I get a awkward problem with cpm_data_transformer.

I0906 10:07:06.670740 20250 cpm_data_transformer.cpp:73] dataset: COCO; img_size: [640 x 427]; meta.annolist_index: 21004; meta.write_number: 0; meta.total_write_number: 121522; meta.epoch: 0
*** Aborted at 1504663626 (unix time) try "date -d @1504663626" if you are using GNU date ***
PC: @     0x7f17a8e31ebb cv::_InputArray::size()
*** SIGSEGV (@0x0) received by PID 20224 (TID 0x7f170effd700) from PID 0; stack trace: ***
    @     0x7f17a7e314b0 (unknown)
    @     0x7f17a8e31ebb cv::_InputArray::size()
    @     0x7f177a511cc3 cv::resize()
    @     0x7f17a9fc011e caffe::CPMDataTransformer<>::augmentation_scale()
    @     0x7f17a9fdc00b caffe::CPMDataTransformer<>::Transform_nv()
    @     0x7f17a9fdccb6 caffe::CPMDataTransformer<>::Transform_nv()
    @     0x7f17a9e55c33 caffe::CPMDataLayer<>::load_batch()
    @     0x7f17a9f03b0f caffe::BasePrefetchingDataLayer<>::InternalThreadEntry()
    @     0x7f17aa029a25 caffe::InternalThread::entry()
    @     0x7f177261e5d5 (unknown)
    @     0x7f176cc5d6ba start_thread
    @     0x7f17a7f033dd clone
    @                0x0 (unknown)

I found the corrupted position at line 644 in cpm_data_transformer.cpp. resize(img_src, img_temp, Size(), scale, scale, INTER_CUBIC) It seemed there were something wrong with the img_src. I tried to save the img_src by imwrite("src.jpg", img_src) before the resize() operation, but it corrupted at imwrite(). I checked the img_src by debugging with gdb, but data, size, channels and type of it were normal. Can anyone give me some advice to resolve this ? Thanks a lot!

[AIHGF]

@wujiyoung Maybe the meta.scale_self = 0,which results in float scale_abs = inf in cpm_data_transformer.cpp - float scale_abs =param_.target_dist()/meta.scale_self.