diff --git a/hens_output b/output/hens_output similarity index 100% rename from hens_output rename to output/hens_output diff --git a/output/output_video.mp4 b/output/output_video.mp4 new file mode 100644 index 0000000000000000000000000000000000000000..4d6eefde2b2ce630c0d5cc5e922f521d629fee58 Binary files /dev/null and b/output/output_video.mp4 differ diff --git a/show_results.py b/show_results.py index 2f66e8c4152adb8dd14ecbfba9dd713d6d316c3c..a9ea2cd9f95e7cbcc48d50c63d76834cbfe03ee4 100644 --- a/show_results.py +++ b/show_results.py @@ -52,14 +52,14 @@ def run(sequence_dir, result_file, show_false_alarms=False, detection_file=None, seq_info["detections"], frame_idx) vis.draw_detections(detections) - mask = results[:, 0].astype(np.int) == frame_idx - track_ids = results[mask, 1].astype(np.int) + mask = results[:, 0].astype(np.int64) == frame_idx + track_ids = results[mask, 1].astype(np.int64) boxes = results[mask, 2:6] vis.draw_groundtruth(track_ids, boxes) if show_false_alarms: groundtruth = seq_info["groundtruth"] - mask = groundtruth[:, 0].astype(np.int) == frame_idx + mask = groundtruth[:, 0].astype(np.int64) == frame_idx gt_boxes = groundtruth[mask, 2:6] for box in boxes: # NOTE(nwojke): This is not strictly correct, because we don't @@ -68,7 +68,7 @@ def run(sequence_dir, result_file, show_false_alarms=False, detection_file=None, if iou(box, gt_boxes).max() < min_iou_overlap: vis.viewer.color = 0, 0, 255 vis.viewer.thickness = 4 - vis.viewer.rectangle(*box.astype(np.int)) + vis.viewer.rectangle(*box.astype(np.int64)) if update_ms is None: update_ms = seq_info["update_ms"] @@ -99,7 +99,7 @@ def parse_args(): default=None) parser.add_argument( "--output_file", help="Filename of the (optional) output video.", - default=None) + default=op_vid) parser.add_argument( "--show_false_alarms", help="Show false alarms as red bounding boxes.", type=bool, default=False)