diff --git a/ieemoo-ai-isempty.py b/ieemoo-ai-isempty.py index 15712b1..a7bb3d0 100644 --- a/ieemoo-ai-isempty.py +++ b/ieemoo-ai-isempty.py @@ -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() + diff --git a/requirements.txt b/requirements.txt old mode 100755 new mode 100644 index 64d403c..ed5c9a3 --- a/requirements.txt +++ b/requirements.txt @@ -1,80 +1,21 @@ -absl-py==1.0.0 -Bottleneck==1.3.2 -brotlipy==0.7.0 -cachetools==5.0.0 -certifi==2021.10.8 -cffi @ file:///tmp/build/80754af9/cffi_1625807838443/work -charset-normalizer @ file:///tmp/build/80754af9/charset-normalizer_1630003229654/work -click==8.0.3 -contextlib2==21.6.0 -cryptography @ file:///tmp/build/80754af9/cryptography_1635366571107/work -cycler @ file:///tmp/build/80754af9/cycler_1637851556182/work -docopt==0.6.2 -esdk-obs-python==3.21.8 -faiss==1.7.1 -Flask @ file:///tmp/build/80754af9/flask_1634118196080/work -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 +esdk_obs_python==3.21.8 +Flask==2.0.0 +gevent==21.1.2 +matplotlib==3.4.1 +numpy==1.20.2 +esdk-obs-python --trusted-host pypi.org +opencv_python==4.5.1.48 +Pillow==9.1.0 +scipy==1.6.2 +setuptools==49.6.0 +ml-collections +coremltools==5.2.0 +onnx==1.7.0 +pandas==1.2.4 +pycocotools==2.0.2 PyYAML==6.0 -requests @ file:///tmp/build/80754af9/requests_1629994808627/work -requests-oauthlib==1.3.1 -rsa==4.8 -scipy @ file:///tmp/build/80754af9/scipy_1630606796110/work -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 +requests==2.25.1 +seaborn==0.11.1 +thop==0.0.31 +tqdm==4.60.0 +