lichen/ieemoo-ai-predict
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 15:35:17 +08:00
parent c30cdd9557
commit 1ea6546e85
2 changed files with 28 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
ieemoo-ai-predict.py
View File

@ -37,9 +37,9 @@ app.logger.addHandler(handler)
load model load model
''' '''
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('--weights', nargs='+', type=str, default='weightsv5s', help='model.pt path(s)') parser.add_argument('--weights', nargs='+', type=str, default='../module/ieemoo-ai-predict/', help='model.pt path(s)')
parser.add_argument('--Form', type=str, default='') parser.add_argument('--Form', type=str, default='')
parser.add_argument('--source', type=str, default='data/images', help='source') # file/folder, 0 for webcam parser.add_argument('--source', type=str, default='../module/ieemoo-ai-predict/imgs', help='source') # file/folder, 0 for webcam
parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)') parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)')
parser.add_argument('--conf-thres', type=float, default=0.25, help='object confidence threshold') parser.add_argument('--conf-thres', type=float, default=0.25, help='object confidence threshold')
parser.add_argument('--iou-thres', type=float, default=0.45, help='IOU threshold for NMS') parser.add_argument('--iou-thres', type=float, default=0.45, help='IOU threshold for NMS')
@ -65,7 +65,8 @@ def loadModel():
device = select_device(opt.device) device = select_device(opt.device)
half = device.type != 'cpu' half = device.type != 'cpu'
for name in os.listdir(opt.weights): for name in os.listdir(opt.weights):
all_model[name.split('.')[0]] = attempt_load(os.sep.join(['weightsv5s', name]), map_location=device) if '.pt' in name:
all_model[name.split('.')[0]] = attempt_load(os.sep.join([opt.weights, name]), map_location=device)
return all_model return all_model
def CheckRange(code): def CheckRange(code):
@ -177,6 +178,7 @@ def predict():
if request.method == 'POST': if request.method == 'POST':
try: try:
data = request.get_data() data = request.get_data()
print(type(data))
data_str = data.decode("utf-8") data_str = data.decode("utf-8")
json_data = json.loads(data_str) json_data = json.loads(data_str)
pic = json_data.get("file") pic = json_data.get("file")
@ -191,7 +193,7 @@ def predict():
section = 'nonentity' section = 'nonentity'
else: else:
model = all_model[section] model = all_model[section]
cv2.imwrite(os.sep.join(['data/images',code+'.jpg']), image) cv2.imwrite(os.sep.join(['../module/ieemoo-ai-predict/imgs',code+'.jpg']), image)
with torch.no_grad(): with torch.no_grad():
if section == 'nonentity': if section == 'nonentity':
tar_res = {'tag':False} tar_res = {'tag':False}
@ -201,11 +203,11 @@ def predict():
#ocr_res = detectOcr(barcode, ocr, 'data/images/'+code+'.jpg') #ocr_res = detectOcr(barcode, ocr, 'data/images/'+code+'.jpg')
ocr_res = {"Ocr": {"barcode_rst": "", "barcode_success": "false", "ocr_rst": "", "ocr_success": False}} ocr_res = {"Ocr": {"barcode_rst": "", "barcode_success": "false", "ocr_rst": "", "ocr_success": False}}
ocr_res.update(tar_res) ocr_res.update(tar_res)
os.remove(os.sep.join(['data/images',code+'.jpg'])) os.remove(os.sep.join(['../module/ieemoo-ai-predict/imgs',code+'.jpg']))
return ocr_res return ocr_res
except: except:
return {"Ocr": {"barcode_rst": "", "barcode_success": "false", "ocr_rst": "", "ocr_success": False}, "TargetDetect": {""}, "tag": False} return {"Ocr": {"barcode_rst": "", "barcode_success": "false", "ocr_rst": "", "ocr_success": False}, "TargetDetect": {""}, "tag": False}
if __name__ == '__main__': if __name__ == '__main__':
#http_server = WSGIServer(('0.0.0.0', 8084), app) http_server = WSGIServer(('0.0.0.0', 8000), app)
#http_server.serve_forever() http_server.serve_forever()
app.run() #app.run()

97
requirements.txt
View File

@ -1,80 +1,19 @@
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==0.1.1
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 tqdm==4.60.0
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