Добавить значение переменной в функцию cv2.imwrite - PullRequest
2 голосов
/ 10 апреля 2019

Я хочу добавить значение строковой переменной в cv2.imwrite() функцию

Я создал код для извлечения кадров из видео и проверки, присутствует ли мой целевой объект в кадре или нет. В настоящее время кадры, в которых найдена цель, сохраняются как frame0, frame1 и так далее. Ниже мой код для того же

def target_non_target(input_frames_folder,model_file):

    count=0
    folders = glob(input_frames_folder)

    img_list = []

    for folder in folders:
        for f in glob(folder+"/*.jpg"):
            img_list.append(f)


    for i in range(len(img_list)):
        print(img_list[i])
        v1=os.path.basename(img_list[i])
        print(v1)
        img_name = os.path.splitext(v1)[0]
        print(img_name)
        image = cv2.imread(img_list[i])
        orig = image.copy()

        image = cv2.resize(image, (28, 28))
        image = image.astype("float") / 255.0
        image = img_to_array(image)
        image = np.expand_dims(image, axis=0)


        print("[INFO] loading network...")
        model = load_model(model_file)


        (non_target, target) = model.predict(image)[0]

        if target > non_target:
            label = "Target"

        else:
            label = "Non Target"

        probab = target if target > non_target else non_target
        label = "{}: {:.2f}%".format(label, probab * 100)


        output = imutils.resize(orig, width=400)
        cv2.putText(output, label, (10, 25),  cv2.FONT_HERSHEY_SIMPLEX,0.7, (0, 255, 0), 2)

        if target > non_target:
            cv2.imwrite("C:\\Python35\\target_non_target\\Target_images_new\\frame%d.jpg"%count,orig)
            count += 1
    #cv2.imshow("Output", output)
        cv2.waitKey(0)



frames_folder = ("C:\\Python36\\videos\\videos_new\\*")

model = ("C:\\Python35\\target_non_target\\target_non_target.model")

target_check = target_non_target(frames_folder,model)

В настоящее время кадры, в которых найден целевой объект, сохраняются как frame0, frame1 и т. Д. Я хочу сохранить кадр с исходным именем изображения, которое входит в переменную IMG_NAME. Как я могу добавить его к функции cv2.imwrite(). Мои предыдущие попытки дали мне эту ошибку

TypeError: не все аргументы преобразованы во время форматирования строки

1 Ответ

0 голосов
/ 10 апреля 2019

Вам необходимо изменить форматирование строки при сохранении кадра с помощью cv2.imwrite().В настоящее время ваше форматирование %d, которое пытается использовать целое число, когда оно должно быть строкой.Вы можете использовать %s или использовать встроенную функцию format () .

Изменить

cv2.imwrite("C:\\Python35\\target_non_target\\Target_images_new\\frame%d.jpg"%count,orig)

на

cv2.imwrite("C:\\Python35\\target_non_target\\Target_images_new\\frame{}.jpg".format(img_name),orig)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...