diff options
| author | Bob Poekert <bob@poekert.com> | 2016-05-13 22:06:33 -0700 |
|---|---|---|
| committer | Bob Poekert <bob@poekert.com> | 2016-05-14 11:55:30 -0700 |
| commit | e8ec9f806bd0051f2ee8d1d2737afdafe314f9e4 (patch) | |
| tree | 9aeb17fdb3f16b316a7f4f4fd1ca37f5d69226d7 /src/caffe/layers/window_data_layer.cpp | |
| parent | bb0c1a5ba18b685dbd62efbac06b2609e20d1d88 (diff) | |
| download | caffeonacl-e8ec9f806bd0051f2ee8d1d2737afdafe314f9e4.tar.gz caffeonacl-e8ec9f806bd0051f2ee8d1d2737afdafe314f9e4.tar.bz2 caffeonacl-e8ec9f806bd0051f2ee8d1d2737afdafe314f9e4.zip | |
add check for background and foreground window size > 0 in WindowData layer
Diffstat (limited to 'src/caffe/layers/window_data_layer.cpp')
| -rw-r--r-- | src/caffe/layers/window_data_layer.cpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/caffe/layers/window_data_layer.cpp b/src/caffe/layers/window_data_layer.cpp index 4ca8315d..103dd4b6 100644 --- a/src/caffe/layers/window_data_layer.cpp +++ b/src/caffe/layers/window_data_layer.cpp @@ -265,6 +265,9 @@ void WindowDataLayer<Dtype>::load_batch(Batch<Dtype>* batch) { const int num_samples[2] = { batch_size - num_fg, num_fg }; int item_id = 0; + CHECK_GT(fg_windows_.size(), 0); + CHECK_GT(bg_windows_.size(), 0); + // sample from bg set then fg set for (int is_fg = 0; is_fg < 2; ++is_fg) { for (int dummy = 0; dummy < num_samples[is_fg]; ++dummy) { |