Fix multi_camera_multi_target_tracking_demo (#3675)

* Fix multi_camera_multi_target_tracking_demo

Close #3480

Close #3619

* Support newer scipy

* multi_camera_multi_target_tracking_demo: test --m_segmentation

Author: Wovchena

demos/multi_camera_multi_target_tracking_demo/python/mc_tracker/sct.py

@@ -511,8 +511,8 @@ def _compute_detections_assignment_cost(self, active_tracks_idx, detections, fea
                 reid_dist_curr, reid_dist_avg, reid_dist_clust = None, None, None
                 if self.tracks[idx].f_avg.is_valid() and features[j] is not None \
                         and self.tracks[idx].get_last_feature() is not None:
-                    reid_dist_avg = 0.5 * cosine(self.tracks[idx].f_avg.get(), features[j])
-                    reid_dist_curr = 0.5 * cosine(self.tracks[idx].get_last_feature(), features[j])
+                    reid_dist_avg = 0.5 * cosine(self.tracks[idx].f_avg.get().squeeze(), features[j].squeeze())
+                    reid_dist_curr = 0.5 * cosine(self.tracks[idx].get_last_feature().squeeze(), features[j].squeeze())
 
                     if self.process_curr_features_number > 0:
                         num_features = len(self.tracks[idx])

demos/multi_camera_multi_target_tracking_demo/python/multi_camera_multi_target_tracking_demo.py

@@ -197,6 +197,7 @@ def run(params, config, capture, detector, reid):
 
     if len(config.embeddings.save_path):
         save_embeddings(tracker.scts, **vars(config.embeddings))
+    detector.infer_queue.wait_all()
 
 
 def main():

demos/multi_camera_multi_target_tracking_demo/python/utils/network_wrappers.py

@@ -125,7 +125,7 @@ def wait_and_grab(self):
         return outputs
 
 
-class MaskRCNN(DetectorInterface):
+class MaskRCNN(IEModel, DetectorInterface):
     """Wrapper class for a network returning masks of objects"""
 
     def __init__(self, core, model_path, trg_classes, conf=.6,
@@ -185,16 +185,16 @@ def forward(self, im_data, im_info):
         if im_info is not None:
             im_info = im_info.reshape(1, *im_info.shape)
             feed_dict['im_info'] = im_info
-        self.infer_request.infer(feed_dict)
+        self.infer_queue[0].infer(feed_dict)
         if self.segmentoly_type:
-            output = {name: self.infer_request.get_tensor(name).data[:] for name in self.output_keys_segmentoly}
+            output = {name: self.infer_queue[0].get_tensor(name).data for name in self.output_keys_segmentoly}
             valid_detections_mask = output['classes'] > 0
             classes = output['classes'][valid_detections_mask]
             boxes = output['boxes'][valid_detections_mask]
             scores = output['scores'][valid_detections_mask]
             masks = output['raw_masks'][valid_detections_mask]
         else:
-            output = {name: self.infer_request.get_tensor(name).data[:] for name in self.output_keys}
+            output = {name: self.infer_queue[0].get_tensor(name).data for name in self.output_keys}
             valid_detections_mask = np.sum(output['boxes'], axis=1) > 0
             classes = output['labels'][valid_detections_mask] + 1
             boxes = output['boxes'][valid_detections_mask][:, :4]

demos/tests/cases.py

@@ -1059,12 +1059,15 @@ def single_option_cases(key, *args):
     )),
 
     PythonDemo(name='multi_camera_multi_target_tracking_demo', device_keys=['-d'],
-               model_keys=['-m', '--m_reid'], test_cases=combine_cases(
+               model_keys=['-m', '--m_segmentation', '--m_reid'], test_cases=combine_cases(
         TestCase(options={'--no_show': None,
             **MONITORS,
             '-i': [DataPatternArg('multi-camera-multi-target-tracking'),
-                DataPatternArg('multi-camera-multi-target-tracking/repeated')],
-            '-m': ModelArg('person-detection-retail-0013')}),
+                DataPatternArg('multi-camera-multi-target-tracking/repeated')]}),
+        [
+            TestCase(options={'-m': ModelArg('person-detection-retail-0013')}),
+            TestCase(options={'--m_segmentation': ModelArg('instance-segmentation-security-0228')}),
+        ],
         single_option_cases('--m_reid',
             ModelArg('person-reidentification-retail-0277'),
             ModelArg('person-reidentification-retail-0286'),