else: boxes = torch.zeros(0) labels = torch.zeros(0) scores = torch.zeros(0) aboxes = torch.zeros(0) alabels = torch.zeros(0) ascores = torch.zeros(0) filename = '_'.join( trainset.ids[index[kk][jj % trainset.time_window]][-1]) print('name: {}, image size: {}x{}x{}'.format( filename, *vis.shape)) draw_boxes(ax, np.array(vis_img), boxes, labels, scores, OBJ_CLASSES) plt.savefig(filename + '_vis.jpg') draw_boxes( ax, 255 * seg[kk, jj % trainset.time_window].squeeze().byte().numpy(), boxes, labels, scores, OBJ_CLASSES) plt.savefig(filename + '_seg.jpg') # seg_img = Image.fromarray(seg[kk, jj% trainset.time_window].squeeze().byte().numpy(), mode='P') # seg_img.putpalette(seg_palette) # seg_img.save( filename + '_seg.png' ) if len(boxes): pdb.set_trace()
alabels = torch.zeros(0) ascores = torch.zeros(0) else: boxes = torch.zeros(0) labels = torch.zeros(0) scores = torch.zeros(0) aboxes = torch.zeros(0) alabels = torch.zeros(0) ascores = torch.zeros(0) filename = '_'.join( trainset.ids[index[kk][jj % trainset.time_window]][-1] ) print('name: {}, image size: {}x{}x{}'.format(filename, *vis.shape)) draw_boxes(ax, np.array(vis_img), boxes, labels, scores, OBJ_CLASSES) plt.savefig( filename + '_vis.jpg' ) # if len(boxes): # gen_mask( boxes ).save( filename + '_mask.png' ) # pdb.set_trace() # draw_boxes(ax, np.array(lwir_img), boxes, labels, scores, OBJ_CLASSES) # plt.savefig( filename + '_lwir.jpg' ) # draw_boxes(ax, np.array(lwir_img), aboxes, alabels, ascores) # plt.savefig( filename + '_lwir_anchors.jpg' ) pdb.set_trace()
# if any(anchor_labels>20): # pdb.set_trace() anchor_hist += np.histogram(anchor_labels.numpy(), np.arange(0, num_anchors + 1))[0] if bSave: filename = '_'.join(trainset.ids[index][-1]) print('name: {}, image size: {}x{}x{}'.format( filename, *vis.shape)) boxes[:, (0, 2)] *= ori_size[1] boxes[:, (1, 3)] *= ori_size[0] draw_boxes(ax, vis, boxes, labels, scores, OBJ_CLASSES, colormap) plt.savefig(filename + '_vis.jpg') draw_boxes(ax, lwir, boxes, labels, scores, OBJ_CLASSES, colormap) plt.savefig(filename + '_lwir.jpg') aboxes = anchors[0] aboxes[:, (0, 2)] *= ori_size[1] aboxes[:, (1, 3)] *= ori_size[0] # draw_boxes(ax, lwir, aboxes, anchors[1]+1, anchors[2]) draw_boxes(ax, lwir, aboxes, anchor_labels, anchors[2],