wko-on-cloud-n/src/setup.py

29 lines
630 B
Python
Raw Normal View History

2022-02-15 04:03:45 +01:00
import os
import numpy as np
import pandas as pd
import random
import matplotlib.pyplot as plt
import time
#progess bar
from tqdm import tqdm
import cv2
import warnings
#deep learning
from consts import SEED, JPG_IMAGES, FC_DIR, FEATURES
from helpers import create_folder, convert_tif_to_jpg
warnings.filterwarnings('ignore')
def transform_photo(tif_dir):
dp = create_folder(tif_dir, JPG_IMAGES)
fc = create_folder(FC_DIR, dp)
convert_tif_to_jpg(os.path.join(FEATURES, tif_dir), dp, fc)
if __name__ == "__main__":
for d in os.listdir(FEATURES):
print(str(d), end='\t')
transform_photo(d)