lichen/ieemoo-ai-isempty
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
lichen
2022-04-12 13:47:32 +08:00
parent f23dc22752
commit 1a4e5b5437
2 changed files with 28 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
ieemoo-ai-isempty.py
View File

@ -7,6 +7,7 @@ import cv2, base64
import argparse
import sys, os
import torch
from gevent.pywsgi import WSGIServer
from PIL import Image
from torchvision import transforms
from models.modeling import VisionTransformer, CONFIGS
@ -17,7 +18,8 @@ app = Flask(__name__)
app.use_reloader=False
def parse_args(model_file="ckpts/emptyjudge5_checkpoint.bin"):
def parse_args(model_file="../module/ieemoo-ai-isempty/emptyjudge5_checkpoint.bin"):
#def parse_args(model_file="output/emptyjudge5_checkpoint.bin"):
parser = argparse.ArgumentParser()
parser.add_argument("--img_size", default=448, type=int, help="Resolution size")
parser.add_argument('--split', type=str, default='overlap', help="Split method")
@ -83,7 +85,8 @@ class Predictor(object):
return result
model_file ="/data/ieemoo/emptypredict_pfc_FG/ckpts/emptyjudge5_checkpoint.bin"
model_file ="../module/ieemoo-ai-isempty/emptyjudge5_checkpoint.bin"
#model_file ="output/emptyjudge5_checkpoint.bin"
args = parse_args(model_file)
predictor = Predictor(args)
@ -118,7 +121,9 @@ def get_isempty():
if __name__ == "__main__":
app.run()
# app.run("0.0.0.0", port=8083)
# http_server = WSGIServer(('0.0.0.0',8000), app)
# http_server.serve_forever()