from flask import Flask, send_file from random import randint from flask import render_template import os from io import BytesIO from image_container import ImageContainer import base64 PATTERN_FOLDER = "static" IMG_DIR = "/home/luud/dataset/deep_fashion" SOURCE_FILE = "/home/luud/dataset/deep_fashion/Anno/list_bbox.txt" DEST_FILE = "" app = Flask(__name__) img_container = ImageContainer(SOURCE_FILE, DEST_FILE, IMG_DIR) def serve_pil_image(pil_img): img_io = BytesIO() pil_img.save(img_io, 'JPEG', quality=70) img_io.seek(0) b64 = base64.b64encode(img_io.read()) return b64 @app.route('/') @app.route('/index') def index(): name = "Annotator" quotes = [ "'If people do not believe that mathematics is simple, it is only because they do not realize how complicated life is.' -- John Louis von Neumann ", "'Computer science is no more about computers than astronomy is about telescopes' -- Edsger Dijkstra ",
# generate_seeds() if len(sys.argv) != 2: raise ValueError("I need 1 arg, the seed filename") seed_filename = sys.argv[1] name = seed_filename.split('/')[-1].split('_')[0] if not os.path.isdir('output'): os.mkdir('output') output_dir = 'output/{}'.format(seed_filename.split('/')[-1].split('.')[0]) if not os.path.isdir(output_dir): os.mkdir('output/{}'.format( seed_filename.split('/')[-1].split('.')[0])) # ############### for using cut images # seed_img = cv.imread(seed_filename) # cut_images_fns = glob.glob("cut_images/{}*".format(name)) # img_containers = [] # for cut_image_fn in cut_images_fns: # img_containers.append(ImageContainer(cv.imread(cut_image_fn), param_type='point')) # ################ for whole images seed_img = cv.imread(seed_filename) img_filename = glob.glob("raw_images/{}*.tif".format(name)) assert len(img_filename) == 1 img_containers = [ ImageContainer(cv.imread(img_filename[0]), param_type='point') ] find_closest(500, seed_img, img_containers, output_dir, step=600)