|
|
@@ -147,7 +147,7 @@ class LineDataset(BaseDataset):
|
|
|
if show_type=='all':
|
|
|
boxed_image = draw_bounding_boxes((img * 255).to(torch.uint8), target["boxes"],
|
|
|
colors="yellow", width=1)
|
|
|
- circle=target['circle']
|
|
|
+ circle=target['circles']
|
|
|
print(f'taget circle:{circle.shape}')
|
|
|
keypoint_img=draw_keypoints(boxed_image,circle,colors='red',width=3)
|
|
|
plt.imshow(keypoint_img.permute(1, 2, 0).numpy())
|
|
|
@@ -257,13 +257,13 @@ def get_boxes_lines(objs,shape):
|
|
|
points=sort_points_clockwise(obj['points'])
|
|
|
circle_4points.append(points)
|
|
|
|
|
|
- xmin = max(obj['xmin'] - 6, 0)
|
|
|
+ xmin = max(obj['xmin'] - 40, 0)
|
|
|
|
|
|
- xmax = min(obj['xmax'] + 6, w)
|
|
|
+ xmax = min(obj['xmax'] + 40, w)
|
|
|
|
|
|
- ymin = max(obj['ymin'] - 6, 0)
|
|
|
+ ymin = max(obj['ymin'] - 40, 0)
|
|
|
|
|
|
- ymax = min(obj['ymax'] + 6, h)
|
|
|
+ ymax = min(obj['ymax'] + 40, h)
|
|
|
|
|
|
boxes.append([xmin, ymin, xmax, ymax])
|
|
|
|
|
|
@@ -303,6 +303,6 @@ def get_boxes_lines(objs,shape):
|
|
|
return boxes,line_point_pairs,points,line_mask,circle_4points, labels
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
- path=r"\\192.168.50.222/share/zyh/data/rgb_4point/a_dataset"
|
|
|
+ path=r'/data/share/zyh/master_dataset/circle/huayan_circle/a_dataset'
|
|
|
dataset= LineDataset(dataset_path=path, dataset_type='train',augmentation=False, data_type='jpg')
|
|
|
- dataset.show(99,show_type='all')
|
|
|
+ dataset.show(9,show_type='all')
|
admin