(output_dir, output_dic, keep)
| 138 | return path |
| 139 | |
| 140 | def save_pred_fig(output_dir, output_dic, keep): |
| 141 | # im = Image.open(os.path.join(output_dir, "img.png")) |
| 142 | im = cv2.imread(os.path.join(output_dir, "img.png")) |
| 143 | h, w = im.shape[:2] |
| 144 | bboxes_scaled = rescale_bboxes(output_dic['pred_boxes'][0, keep].cpu(), (w,h)) |
| 145 | prob = output_dic['pred_logits'].softmax(-1)[0, :, :-1] |
| 146 | scores = prob[keep] |
| 147 | plot_results(im, scores, bboxes_scaled, output_dir) |
| 148 | |
| 149 | def save_gt_fig(output_dir, gt_anno): |
| 150 | im = cv2.imread(os.path.join(output_dir, "img.png")) |
no test coverage detected