Как прочитать каждый текст, преобразовать его в изображение и сохранить в разных изображениях? - PullRequest
0 голосов
/ 06 мая 2020

Я хотел бы, чтобы мой код считывал мой текстовый файл один за другим, преобразовывал его в изображение и сохранял с другим изображением, то есть 300, 600, 900, ....

Я сделал код вниз, и он говорит только путь, но не переходит к следующему коду.

Не могли бы вы дать мне совет или найти недостающую или ошибочную часть моих кодов?

import numpy as np
import matplotlib.pyplot as plt
import glob
import cv2
import os


path = './Master_thesis/Code/dnn_simulation_result/'
interval = 300

folders = []

#r=root, d=dirctories, f=files

for r, d, f in os.walk(path):
    if not d:
        folders.append(r)

for f in folders:
    print(r)


def txt2image(folders, skiprows) :    

    for folder_name in folders:

        IsFile=(glob.glob(folder_name+"/*.*"))

        for file in IsFile:

            myArray  = np.loadtxt(path, skiprows = skiprows)

            # Set the nodata values to nan
            myArray[myArray == -9999] = np.nan

            # PRISM data is stored as an integer but scaled by 100
            myArray *= 1

            # Plot PRISM array again
            fig, ax = plt.subplots()
            ax.set_title('Flood area')

            # Get the img object in order to pass it to the colorbar function
            img_plot = ax.imshow(myArray, cmap='jet')

            # Place a colorbar next to the map
            cbar = fig.colorbar(img_plot)

            ax.grid(True)
            plt.show()        

            txt2image = cv2.imwrite('D:/Master_thesis/Code/dnn_simulation_result/dnn_simulation_result/{}.jpg', img_plot)

        return txt2image

txt2image(folders, 0)

enter image description here

...