diff --git a/nets/yolo_training.py b/nets/yolo_training.py index eb83296c9c9a6053a225c219cbb0585e37ba3663..441f11d393214356046640d4feb911c2dbae4178 100644 --- a/nets/yolo_training.py +++ b/nets/yolo_training.py @@ -418,7 +418,7 @@ class Generator(object): dy = place_y[index] new_image = Image.new('RGB', (w,h), (128,128,128)) new_image.paste(image, (dx, dy)) - image_data = np.array(new_image)/255 + image_data = np.array(new_image) index = index + 1