(image, box, caption, color)
| 128 | return matches, unmatched1, unmatched2 |
| 129 | |
| 130 | def draw_caption(image, box, caption, color): |
| 131 | b = np.array(box).astype(int) |
| 132 | cv2.putText(image, caption, (b[0], b[1] - 8), cv2.FONT_HERSHEY_PLAIN, 2, color, 2) |
| 133 | |
| 134 | |
| 135 | def run_each_dataset(model_dir, retinanet, dataset_path, subset, cur_dataset): |