Set overlaps_ignore

This commit is contained in:
Juha Reunanen 2019-10-26 20:55:54 +03:00
parent 075d24afdd
commit 65adeff1f8

View File

@ -190,6 +190,9 @@ det_bnet_type train_detection_network(
const double momentum = 0.9;
mmod_options options(mmod_rects, 70, 30);
options.overlaps_ignore = test_box_overlap(0.5, 0.95);
det_bnet_type det_net(options);
det_net.subnet().layer_details().set_num_filters(options.detector_windows.size());