Skip to content

Commit 5f21dad

Browse files
committed
Merge pull request BVLC#4148 from bobpoekert/window_data_nonzero_check
Window data foreground/background nonzero check
2 parents 7cf3538 + e8ec9f8 commit 5f21dad

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/caffe/layers/window_data_layer.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -265,6 +265,9 @@ void WindowDataLayer<Dtype>::load_batch(Batch<Dtype>* batch) {
265265
const int num_samples[2] = { batch_size - num_fg, num_fg };
266266

267267
int item_id = 0;
268+
CHECK_GT(fg_windows_.size(), 0);
269+
CHECK_GT(bg_windows_.size(), 0);
270+
268271
// sample from bg set then fg set
269272
for (int is_fg = 0; is_fg < 2; ++is_fg) {
270273
for (int dummy = 0; dummy < num_samples[is_fg]; ++dummy) {

0 commit comments

Comments
 (0)