Commit 17e0d318c059c7260321409367dc04db878153cc
1 parent
52484e12
Block project only when this transform is trained
Showing
1 changed file
with
8 additions
and
3 deletions
openbr/plugins/slidingwindow.cpp
| ... | ... | @@ -94,8 +94,10 @@ private: |
| 94 | 94 | void project(const Template &src, Template &dst) const |
| 95 | 95 | { |
| 96 | 96 | dst = src; |
| 97 | - // no need to slide a window over ground truth data | |
| 98 | - if (src.file.getBool("Train", false)) return; | |
| 97 | + if (skipProject) { | |
| 98 | + dst = src; | |
| 99 | + return; | |
| 100 | + } | |
| 99 | 101 | |
| 100 | 102 | dst.file.clearRects(); |
| 101 | 103 | float scale = src.file.get<float>("scale", 1); |
| ... | ... | @@ -255,7 +257,10 @@ private: |
| 255 | 257 | void project(const Template &src, Template &dst) const |
| 256 | 258 | { |
| 257 | 259 | dst = src; |
| 258 | - // do not scale images during training | |
| 260 | + if (skipProject) { | |
| 261 | + dst = src; | |
| 262 | + return; | |
| 263 | + } | |
| 259 | 264 | if (src.file.getBool("Train", false)) return; |
| 260 | 265 | |
| 261 | 266 | int rows = src.m().rows; | ... | ... |