File size: 1,391 Bytes
00c3521 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
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:
# wb vector
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')
|