File tree Expand file tree Collapse file tree 2 files changed +109
-4
lines changed
Expand file tree Collapse file tree 2 files changed +109
-4
lines changed Original file line number Diff line number Diff line change @@ -962,14 +962,41 @@ PReLU(
962962 shared_axes=None,
963963 **kwargs
964964)
965+ RandAugment(
966+ value_range=(0, 255),
967+ num_ops=2,
968+ factor=0.5,
969+ interpolation='bilinear',
970+ seed=None,
971+ data_format=None,
972+ **kwargs
973+ )
965974RandomBrightness(
966975 factor,
967976 value_range=(0, 255),
968977 seed=None,
969978 **kwargs
970979)
980+ RandomColorDegeneration(
981+ factor,
982+ value_range=(0, 255),
983+ data_format=None,
984+ seed=None,
985+ **kwargs
986+ )
987+ RandomColorJitter(
988+ value_range=(0, 255),
989+ brightness_factor=None,
990+ contrast_factor=None,
991+ saturation_factor=None,
992+ hue_factor=None,
993+ seed=None,
994+ data_format=None,
995+ **kwargs
996+ )
971997RandomContrast(
972998 factor,
999+ value_range=(0, 255),
9731000 seed=None,
9741001 **kwargs
9751002)
@@ -987,6 +1014,26 @@ RandomFlip(
9871014 data_format=None,
9881015 **kwargs
9891016)
1017+ RandomGrayscale(
1018+ factor=0.5,
1019+ data_format=None,
1020+ seed=None,
1021+ **kwargs
1022+ )
1023+ RandomHue(
1024+ factor,
1025+ value_range=(0, 255),
1026+ data_format=None,
1027+ seed=None,
1028+ **kwargs
1029+ )
1030+ RandomPosterization(
1031+ factor,
1032+ value_range=(0, 255),
1033+ data_format=None,
1034+ seed=None,
1035+ **kwargs
1036+ )
9901037RandomRotation(
9911038 factor,
9921039 fill_mode='reflect',
@@ -996,6 +1043,30 @@ RandomRotation(
9961043 data_format=None,
9971044 **kwargs
9981045)
1046+ RandomSaturation(
1047+ factor,
1048+ value_range=(0, 255),
1049+ data_format=None,
1050+ seed=None,
1051+ **kwargs
1052+ )
1053+ RandomSharpness(
1054+ factor,
1055+ value_range=(0, 255),
1056+ data_format=None,
1057+ seed=None,
1058+ **kwargs
1059+ )
1060+ RandomShear(
1061+ x_factor=0.0,
1062+ y_factor=0.0,
1063+ interpolation='bilinear',
1064+ fill_mode='reflect',
1065+ fill_value=0.0,
1066+ data_format=None,
1067+ seed=None,
1068+ **kwargs
1069+ )
9991070RandomTranslation(
10001071 height_factor,
10011072 width_factor,
Original file line number Diff line number Diff line change @@ -10,13 +10,27 @@ affine_transform(
1010 width,
1111 center_x=None,
1212 center_y=None,
13- format ='xyxy'
13+ bounding_box_format ='xyxy'
1414)
1515clip_to_image_size(
1616 bounding_boxes,
1717 height=None,
1818 width=None,
19- format='xyxy'
19+ bounding_box_format='xyxy'
20+ )
21+ compute_ciou(
22+ boxes1,
23+ boxes2,
24+ bounding_box_format,
25+ image_shape=None
26+ )
27+ compute_iou(
28+ boxes1,
29+ boxes2,
30+ bounding_box_format,
31+ use_masking=False,
32+ mask_val=-1,
33+ image_shape=None
2034)
2135convert_format(
2236 boxes,
@@ -32,12 +46,32 @@ crop(
3246 left,
3347 height,
3448 width,
35- format='xyxy'
49+ bounding_box_format='xyxy'
50+ )
51+ decode_deltas_to_boxes(
52+ anchors,
53+ boxes_delta,
54+ anchor_format,
55+ box_format,
56+ encoded_format='center_yxhw',
57+ variance=None,
58+ image_shape=None
59+ )
60+ encode_box_to_deltas(
61+ anchors,
62+ boxes,
63+ anchor_format,
64+ box_format,
65+ encoding_format='center_yxhw',
66+ variance=None,
67+ image_shape=None
3668)
3769pad(
3870 boxes,
3971 top,
4072 left,
41- format='xyxy'
73+ height=None,
74+ width=None,
75+ bounding_box_format='xyxy'
4276)
4377
You can’t perform that action at this time.
0 commit comments