Список из переменных Python 3.6 - PullRequest
Новая
       31

Список из переменных Python 3.6

0 голосов
/ 14 декабря 2018

В основном я хочу составить список из двух переменных, которые меняются.Я использую этот код от Адриана Роузброка, чтобы найти дисперсию лапласиана на изображении, но я хочу, чтобы при смене рисунка он составлял список из двух столбцов "имя_файла" "fm".Вот текущий код скрипта 

from imutils import paths
import argparse
import cv2

def variance_of_laplacian(image):
    # compute the Laplacian of the image and then return the focus
    # measure, which is simply the variance of the Laplacian
    return cv2.Laplacian(image, cv2.CV_64F).var()

# construct the argument parse and parse the arguments
ap = argparse.ArgumentParser()
ap.add_argument("-i", "--images", required=True,
    help="path to input directory of images")
ap.add_argument("-t", "--threshold", type=float, default=100.0,
    help="focus measures that fall below this value will be considered 'blurry'")
args = vars(ap.parse_args())

# loop over the input images
for imagePath in paths.list_images(args["images"]):
    # load the image, convert it to grayscale, and compute the
    # focus measure of the image using the Variance of Laplacian
    # method
    image = cv2.imread(imagePath)
    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
    fm = variance_of_laplacian(gray)
    text = "Not Blurry"

    # if the focus measure is less than the supplied threshold,
    # then the image should be considered "blurry"
    if fm < args["threshold"]:
        text = "Blurry"

    # show the image
    cv2.putText(image, "{}: {:.2f}".format(text, fm), (10, 30),
        cv2.FONT_HERSHEY_SIMPLEX, 0.8, (0, 0, 255), 3)
    cv2.imshow("Image", image)
    key = cv2.waitKey(0)

Так что я хочу, чтобы он сделал .txt или еще что-нибудь с именем файла и значением fm.Надеюсь, вы понимаете, ребята!Спасибо

1 Ответ

0 голосов
/ 14 декабря 2018

Чтобы записать результаты в файл, вы захотите сделать что-то вроде этого: 

# .... skipping initial setup
with open("results.txt", "w") as results_file:
    for imagePath in paths.list_images(args["images"]):
        # .... skip image analysis, results stored in variable 'text'
        # write result to file
        results_file.write("{} {}\n".format(imagePath, text))
# results_file will be closed when the with loop ends
...