run_inference.py 文件源码

def url():
  # check if the post request has the file part
  url = request.query_string[2:]
  if url == "":
    #print("not url")
    return redirect("/")
  title = uuid.uuid4().hex
  path = os.path.join(app.config['UPLOAD_FOLDER'], title)

  # get url file
  (filename, headers) = urllib.urlretrieve(url, path)
  # convert to jpg
  im = Image.open(path)
  if im.mode != "RGB":
    im = im.convert("RGB")
  im.save(path+".jpg", "JPEG")

  FLAGS.input_files = path+".jpg"
  # inference
  result = main(None)
  return jsonify(result)