update
This commit is contained in:
lichen
@ -7,6 +7,7 @@ import cv2, base64
|
|||||||
import argparse
|
import argparse
|
||||||
import sys, os
|
import sys, os
|
||||||
import torch
|
import torch
|
||||||
|
from gevent.pywsgi import WSGIServer
|
||||||
from PIL import Image
|
from PIL import Image
|
||||||
from torchvision import transforms
|
from torchvision import transforms
|
||||||
from models.modeling import VisionTransformer, CONFIGS
|
from models.modeling import VisionTransformer, CONFIGS
|
||||||
@ -17,7 +18,8 @@ app = Flask(__name__)
|
|||||||
app.use_reloader=False
|
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 = argparse.ArgumentParser()
|
||||||
parser.add_argument("--img_size", default=448, type=int, help="Resolution size")
|
parser.add_argument("--img_size", default=448, type=int, help="Resolution size")
|
||||||
parser.add_argument('--split', type=str, default='overlap', help="Split method")
|
parser.add_argument('--split', type=str, default='overlap', help="Split method")
|
||||||
@ -83,7 +85,8 @@ class Predictor(object):
|
|||||||
return result
|
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)
|
args = parse_args(model_file)
|
||||||
predictor = Predictor(args)
|
predictor = Predictor(args)
|
||||||
|
|
||||||
@ -118,7 +121,9 @@ def get_isempty():
|
|||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
app.run()
|
app.run()
|
||||||
# app.run("0.0.0.0", port=8083)
|
# http_server = WSGIServer(('0.0.0.0',8000), app)
|
||||||
|
# http_server.serve_forever()
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
99
requirements.txt
Executable file → Normal file
99
requirements.txt
Executable file → Normal file
@ -1,80 +1,21 @@
|
|||||||
absl-py==1.0.0
|
esdk_obs_python==3.21.8
|
||||||
Bottleneck==1.3.2
|
Flask==2.0.0
|
||||||
brotlipy==0.7.0
|
gevent==21.1.2
|
||||||
cachetools==5.0.0
|
matplotlib==3.4.1
|
||||||
certifi==2021.10.8
|
numpy==1.20.2
|
||||||
cffi @ file:///tmp/build/80754af9/cffi_1625807838443/work
|
esdk-obs-python --trusted-host pypi.org
|
||||||
charset-normalizer @ file:///tmp/build/80754af9/charset-normalizer_1630003229654/work
|
opencv_python==4.5.1.48
|
||||||
click==8.0.3
|
Pillow==9.1.0
|
||||||
contextlib2==21.6.0
|
scipy==1.6.2
|
||||||
cryptography @ file:///tmp/build/80754af9/cryptography_1635366571107/work
|
setuptools==49.6.0
|
||||||
cycler @ file:///tmp/build/80754af9/cycler_1637851556182/work
|
ml-collections
|
||||||
docopt==0.6.2
|
coremltools==5.2.0
|
||||||
esdk-obs-python==3.21.8
|
onnx==1.7.0
|
||||||
faiss==1.7.1
|
pandas==1.2.4
|
||||||
Flask @ file:///tmp/build/80754af9/flask_1634118196080/work
|
pycocotools==2.0.2
|
||||||
fonttools==4.25.0
|
|
||||||
gevent @ file:///tmp/build/80754af9/gevent_1628273677693/work
|
|
||||||
google-auth==2.6.0
|
|
||||||
google-auth-oauthlib==0.4.6
|
|
||||||
greenlet @ file:///tmp/build/80754af9/greenlet_1628887725296/work
|
|
||||||
grpcio==1.44.0
|
|
||||||
gunicorn==20.1.0
|
|
||||||
h5py @ file:///tmp/build/80754af9/h5py_1637138879700/work
|
|
||||||
idna @ file:///tmp/build/80754af9/idna_1637925883363/work
|
|
||||||
importlib-metadata==4.11.3
|
|
||||||
itsdangerous @ file:///tmp/build/80754af9/itsdangerous_1621432558163/work
|
|
||||||
Jinja2 @ file:///tmp/build/80754af9/jinja2_1635780242639/work
|
|
||||||
kiwisolver @ file:///tmp/build/80754af9/kiwisolver_1612282420641/work
|
|
||||||
Markdown==3.3.6
|
|
||||||
MarkupSafe @ file:///tmp/build/80754af9/markupsafe_1621528148836/work
|
|
||||||
matplotlib @ file:///tmp/build/80754af9/matplotlib-suite_1638289681807/work
|
|
||||||
mkl-fft==1.3.1
|
|
||||||
mkl-random @ file:///tmp/build/80754af9/mkl_random_1626186064646/work
|
|
||||||
mkl-service==2.4.0
|
|
||||||
ml-collections==0.1.0
|
|
||||||
munkres==1.1.4
|
|
||||||
numexpr @ file:///tmp/build/80754af9/numexpr_1618856167419/work
|
|
||||||
numpy @ file:///tmp/build/80754af9/numpy_and_numpy_base_1634095647912/work
|
|
||||||
oauthlib==3.2.0
|
|
||||||
olefile @ file:///Users/ktietz/demo/mc3/conda-bld/olefile_1629805411829/work
|
|
||||||
opencv-python==4.5.4.60
|
|
||||||
packaging @ file:///tmp/build/80754af9/packaging_1637314298585/work
|
|
||||||
pandas==1.3.4
|
|
||||||
Pillow==8.4.0
|
|
||||||
pipreqs==0.4.11
|
|
||||||
protobuf==3.19.4
|
|
||||||
pyasn1==0.4.8
|
|
||||||
pyasn1-modules==0.2.8
|
|
||||||
pycparser @ file:///tmp/build/80754af9/pycparser_1636541352034/work
|
|
||||||
pycryptodome==3.10.1
|
|
||||||
pyOpenSSL @ file:///tmp/build/80754af9/pyopenssl_1635333100036/work
|
|
||||||
pyparsing @ file:///tmp/build/80754af9/pyparsing_1635766073266/work
|
|
||||||
PySocks @ file:///tmp/build/80754af9/pysocks_1605305779399/work
|
|
||||||
python-dateutil @ file:///tmp/build/80754af9/python-dateutil_1626374649649/work
|
|
||||||
pytz==2021.3
|
|
||||||
PyYAML==6.0
|
PyYAML==6.0
|
||||||
requests @ file:///tmp/build/80754af9/requests_1629994808627/work
|
requests==2.25.1
|
||||||
requests-oauthlib==1.3.1
|
seaborn==0.11.1
|
||||||
rsa==4.8
|
thop==0.0.31
|
||||||
scipy @ file:///tmp/build/80754af9/scipy_1630606796110/work
|
tqdm==4.60.0
|
||||||
seaborn @ file:///tmp/build/80754af9/seaborn_1629307859561/work
|
|
||||||
sip==4.19.13
|
|
||||||
six @ file:///tmp/build/80754af9/six_1623709665295/work
|
|
||||||
supervisor==4.2.2
|
|
||||||
tensorboard==2.8.0
|
|
||||||
tensorboard-data-server==0.6.1
|
|
||||||
tensorboard-plugin-wit==1.8.1
|
|
||||||
torch==1.8.0
|
|
||||||
torchaudio==0.8.0a0+a751e1d
|
|
||||||
torchvision==0.9.0
|
|
||||||
tornado @ file:///tmp/build/80754af9/tornado_1606942300299/work
|
|
||||||
tqdm @ file:///tmp/build/80754af9/tqdm_1635330843403/work
|
|
||||||
typing-extensions @ file:///tmp/build/80754af9/typing_extensions_1631814937681/work
|
|
||||||
urllib3==1.26.7
|
|
||||||
Werkzeug @ file:///tmp/build/80754af9/werkzeug_1635505089296/work
|
|
||||||
yacs @ file:///tmp/build/80754af9/yacs_1634047592950/work
|
|
||||||
yarg==0.1.9
|
|
||||||
zipp==3.7.0
|
|
||||||
zope.event==4.5.0
|
|
||||||
zope.interface @ file:///tmp/build/80754af9/zope.interface_1625035545636/work
|
|
||||||
|
|||||||
Reference in New Issue
Block a user