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 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
View 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