diff options
| -rw-r--r-- | README.md | 5 | ||||
| -rw-r--r-- | megapixels/app/server/api.py | 2 |
2 files changed, 4 insertions, 3 deletions
@@ -14,8 +14,9 @@ conda install faiss-cpu -c pytorch pip install numpy Pillow pip install dlib pip install requests simplejson click pdfminer.six -pip install urllib3 flask flask_sqlalchemy -pip install pymediainfo tqdm opencv-python imutils scikit-image python-dotenv imagehash scikit-learn colorlog +pip install urllib3 flask flask_sqlalchemy mysql-connector +pip install pymediainfo tqdm opencv-python imutils +pip install scikit-image python-dotenv imagehash scikit-learn colorlog sudo apt-get install libmysqlclient-dev diff --git a/megapixels/app/server/api.py b/megapixels/app/server/api.py index 9c874d81..8ff06611 100644 --- a/megapixels/app/server/api.py +++ b/megapixels/app/server/api.py @@ -106,7 +106,7 @@ def upload(name): for _d, _i in zip(distances, indexes): if _d <= THRESHOLD: dists.append(round(float(_d), 2)) - ids.append(_i+1) + ids.append(_i) results = [ dataset.get_identity(int(_i)) for _i in ids ] |