|
from PIL import Image |
|
import json |
|
import os |
|
|
|
def json_read(fname, **kwargs): |
|
with open(fname) as j: |
|
data = json.load(j, **kwargs) |
|
return data |
|
|
|
|
|
def json_save(fname, data, indent_len=4, **kwargs): |
|
with open(fname, "w") as f: |
|
s = json.dumps(data, sort_keys=True, ensure_ascii=False, |
|
indent=" " * indent_len, **kwargs) |
|
f.write(s) |
|
|
|
|
|
def process_wb_from_txt(txt_path): |
|
with open(txt_path, 'r') as fh: |
|
txt = [line.rstrip().split() for line in fh] |
|
|
|
txt = [[float(k) for k in row] for row in txt] |
|
|
|
assert len(txt) in [1, 3] |
|
|
|
if len(txt) == 1: |
|
|
|
txt = txt[0] |
|
|
|
return txt |
|
|
|
|
|
def process_ids_from_txt(txt_path): |
|
with open(txt_path, 'r') as fh: |
|
temp = fh.read().splitlines() |
|
return temp |
|
|
|
|
|
def save_txt(p, s): |
|
with open(p, 'w') as text_file: |
|
text_file.write(s) |
|
|
|
|
|
def downscale_jpg(img_path, new_shape, quality_perc=100): |
|
img = Image.open(img_path) |
|
if (img.size[0], img.size[1]) != new_shape: |
|
new_img = img.resize(new_shape, Image.ANTIALIAS) |
|
new_img.save(img_path[:-len('.jpg')] + '.jpg', |
|
'JPEG', quality=quality_perc) |
|
|
|
|
|
def rename_img(img_path): |
|
if img_path.lower().endswith('jpeg'): |
|
os.rename(img_path, img_path[:-len('jpeg')] + 'jpg') |
|
else: |
|
os.rename(img_path, img_path[:-len('JPG')] + 'jpg') |
|
|
