from keras.layers import Input
from ssd import SSD
from PIL import Image
ssd = SSD()
while True:
img = input('Input image filename:')
try:
image = Image.open(img)
except:
print('Open Error! Try again!')
continue
else:
r_image = ssd.detect_image(image)
r_image.show()
ssd.close_session()
