Вывод обнаружения объектов для изображений с высоким разрешением занимает много времени на процессоре - PullRequest
Новая
       63

Вывод обнаружения объектов для изображений с высоким разрешением занимает много времени на процессоре

0 голосов
/ 09 июня 2019

Я обучил модель ML для набора данных паскаля с размером изображения 224, но при выводе на новые изображения (некоторые имеют высокое разрешение, а некоторые имеют немного более высокое разрешение, чем изображения паскаль), я получаю ошибку в pil2tensor()

@app.route('/analyze', methods=['POST'])
async def analyze(request):
    data = await request.form()
    img_bytes = await (data['file'].read())
    img = open_image(BytesIO(img_bytes))

    t_img= PIL.Image.open(BytesIO(img_bytes)).convert('RGB')
    t_img = pil2tensor(t_img, np.float32)
    t_img = t_img.div_(255)
    with torch.no_grad():
        # test_output = learn.model.eval()(t_img.unsqueeze_(0).cuda())
        test_output = learn.model.eval()(t_img.unsqueeze_(0))

для изображений небольшого размера (например, некоторых изображений с низким разрешением от Google), модель ML правильно делает вывод за считанные секунды, но для изображений с чуть более высоким разрешением это занимает около 20-40 минут !!!

...