OrgLion-ML
/
dlib
mirror of https://github.com/davisking/dlib.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Really train only instances of the selected class

This commit is contained in:
Juha Reunanen 2019-10-28 07:16:36 +02:00
parent a841eff8a9
commit 6025fa3ffd
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
examples/dnn_instance_segmentation_train_ex.cpp
View File

@ -552,8 +552,16 @@ void filter_listing(
); );
if (has_desired_class) { if (has_desired_class) {
std::vector<truth_instance> temp;
std::copy_if(
truth_instances[i].begin(),
truth_instances[i].end(),
std::back_inserter(temp),
represents_desired_class
);
filtered_listing.push_back(listing[i]); filtered_listing.push_back(listing[i]);
filtered_truth_instances.push_back(truth_instances[i]); filtered_truth_instances.push_back(temp);
} }
} }
@ -583,7 +591,7 @@ int main(int argc, char** argv) try
} }
// mini-batches smaller than the default can be used with GPUs having less memory // mini-batches smaller than the default can be used with GPUs having less memory
const unsigned int det_minibatch_size = argc >= 3 ? std::stoi(argv[2]) : 60; const unsigned int det_minibatch_size = argc >= 3 ? std::stoi(argv[2]) : 75;
const unsigned int seg_minibatch_size = argc >= 4 ? std::stoi(argv[3]) : 25; const unsigned int seg_minibatch_size = argc >= 4 ? std::stoi(argv[3]) : 25;
cout << "det mini-batch size: " << det_minibatch_size << endl; cout << "det mini-batch size: " << det_minibatch_size << endl;
cout << "seg mini-batch size: " << seg_minibatch_size << endl; cout << "seg mini-batch size: " << seg_minibatch_size << endl;