diff --git a/generate_training_data.m b/generate_training_data.m
index 7bf0941..fc01911 100644
--- a/generate_training_data.m
+++ b/generate_training_data.m
@@ -125,6 +125,7 @@
         dres = dres_train{i};
         [~, ~, overlap] = calc_overlap(dres_pole, 1, dres, 1:numel(dres.fr));
         dres.covered = max([dres.covered, overlap'], [], 2);
+        dres.occluded(dres.covered > opt.overlap_occ) = 1;
         dres_train{i} = dres;
     end
 end