|
|
@@ -14,7 +14,6 @@ import time
|
|
|
from io import BytesIO
|
|
|
|
|
|
import boto3
|
|
|
-import cv2
|
|
|
import numpy as np
|
|
|
from PIL import Image
|
|
|
|
|
|
@@ -55,7 +54,6 @@ class SageMakerAiObject:
|
|
|
img_list = map(BytesIO, img_list)
|
|
|
img_list = map(Image.open, img_list)
|
|
|
img_list = map(np.array, img_list)
|
|
|
- img_list = map(lambda x: cv2.cvtColor(x, cv2.COLOR_RGB2BGR), img_list)
|
|
|
img_list = np.array(list(img_list))
|
|
|
|
|
|
nms_threshold = 0.45
|
zhangdongming