File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -78,7 +78,9 @@ def __init__(
7878 self .learning_rate = learning_rate
7979 self .max_iter = max_iter
8080 self .batch_size = batch_size
81- self ._patch = np .random .randint (self .estimator .clip_values [0 ], self .estimator .clip_values [1 ], size = patch_shape ).astype (np .float32 )
81+ self ._patch = np .random .randint (
82+ self .estimator .clip_values [0 ], self .estimator .clip_values [1 ], size = patch_shape
83+ ).astype (np .float32 )
8284 self ._check_params ()
8385
8486 self .target_label = []
@@ -136,8 +138,8 @@ def generate(
136138
137139 target_dict = dict ()
138140 target_dict ["boxes" ] = np .asarray ([[i_x_1 , i_y_1 , i_x_2 , i_y_2 ]])
139- target_dict ["labels" ] = np .asarray ([self .target_label [i_image ], ])
140- target_dict ["scores" ] = np .asarray ([1.0 , ])
141+ target_dict ["labels" ] = np .asarray ([self .target_label [i_image ],])
142+ target_dict ["scores" ] = np .asarray ([1.0 ,])
141143
142144 patch_target .append (target_dict )
143145
You can’t perform that action at this time.
0 commit comments