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)
