@@ -80,17 +80,21 @@ module.exports = () => {
80
80
'TrackerTLD'
81
81
] ;
82
82
83
- if ( cv . version . minor > 0 ) {
83
+ const hasCSRT = ( cv . version . minor > 4 || ( cv . version . minor === 4 && cv . version . patch > 0 ) ) ;
84
+ const hasMOSSE = ( cv . version . minor > 3 ) ;
85
+ const hasKCF = ( cv . version . minor > 0 ) ;
86
+
87
+ if ( hasKCF ) {
84
88
trackerNames . push ( 'TrackerKCF' ) ;
85
89
}
86
90
87
91
if ( cv . version . minor > 1 ) {
88
92
// trackerNames.push('TrackerGOTURN'); TODO: sample goturn.prototxt
89
93
}
90
- if ( cv . version . minor > 4 || ( cv . version . minor === 4 && cv . version . patch > 0 ) ) {
94
+ if ( hasCSRT ) {
91
95
trackerNames . push ( 'TrackerCSRT' ) ;
92
96
}
93
- if ( cv . version . minor > 3 ) {
97
+ if ( hasMOSSE ) {
94
98
trackerNames . push ( 'TrackerMOSSE' ) ;
95
99
}
96
100
trackerNames . forEach ( ( trackerName ) => {
@@ -128,40 +132,43 @@ module.exports = () => {
128
132
const ret = tracker . addKCF ( testImg , new cv . Rect ( 0 , 0 , 10 , 10 ) ) ;
129
133
expect ( ret ) . to . true ;
130
134
} ) ;
131
-
132
- it ( 'addCSRT' , ( ) => {
133
- const tracker = new cv . MultiTracker ( ) ;
134
- const ret = tracker . addCSRT ( testImg , new cv . Rect ( 0 , 0 , 10 , 10 ) ) ;
135
- expect ( ret ) . to . true ;
136
- } ) ;
137
-
138
- it ( 'addMOSSE' , ( ) => {
139
- const tracker = new cv . MultiTracker ( ) ;
140
- const ret = tracker . addMOSSE ( testImg , new cv . Rect ( 0 , 0 , 10 , 10 ) ) ;
141
- expect ( ret ) . to . true ;
142
- } ) ;
135
+ if ( hasCSRT ) {
136
+ it ( 'addCSRT' , ( ) => {
137
+ const tracker = new cv . MultiTracker ( ) ;
138
+ const ret = tracker . addCSRT ( testImg , new cv . Rect ( 0 , 0 , 10 , 10 ) ) ;
139
+ expect ( ret ) . to . true ;
140
+ } ) ;
141
+ }
142
+
143
+ if ( hasMOSSE ) {
144
+ it ( 'addMOSSE' , ( ) => {
145
+ const tracker = new cv . MultiTracker ( ) ;
146
+ const ret = tracker . addMOSSE ( testImg , new cv . Rect ( 0 , 0 , 10 , 10 ) ) ;
147
+ expect ( ret ) . to . true ;
148
+ } ) ;
149
+ }
143
150
} ) ;
144
151
145
152
describe ( 'update' , ( ) => {
146
153
funcShouldRequireArgs ( ( ) => ( new cv . MultiTracker ( ) ) . update ( ) ) ;
147
154
148
155
it ( 'returns bounding box' , ( ) => {
149
156
const tracker = new cv . MultiTracker ( ) ;
150
- const methods = [ 'addMIL' , 'addBOOSTING' , 'addMEDIANFLOW' , 'addTLD' , 'addKCF' , 'addCSRT' , 'addMOSSE' ] ;
151
- if ( cv . version . minor > 0 ) {
157
+ const methods = [ 'addMIL' , 'addBOOSTING' , 'addMEDIANFLOW' , 'addTLD' , 'addKCF' ] ;
158
+ if ( hasKCF ) {
152
159
methods . push ( 'addKCF' ) ;
153
160
}
154
161
155
162
// if (cv.version.minor > 1) {
156
163
// methods.push('addGOTURN');
157
164
// }
158
- if ( cv . version . minor > 4 || ( cv . version . minor === 4 && cv . version . patch > 0 ) ) {
165
+ if ( hasCSRT ) {
159
166
methods . push ( 'addCSRT' ) ;
160
167
}
161
- if ( cv . version . minor > 3 ) {
168
+ if ( hasMOSSE ) {
162
169
methods . push ( 'addMOSSE' ) ;
163
170
}
164
-
171
+
165
172
methods . forEach ( ( addMethod ) => {
166
173
tracker [ addMethod ] ( testImg , new cv . Rect ( 0 , 0 , 10 , 10 ) ) ;
167
174
} ) ;
0 commit comments