From 902d6af0af1ea36b01a346a5c92bd63bb553fd8c Mon Sep 17 00:00:00 2001 From: Zengyf-CVer <41098760+Zengyf-CVer@users.noreply.github.com> Date: Wed, 30 Mar 2022 16:08:02 +0800 Subject: [PATCH 1/5] Update restapi.py --- utils/flask_rest_api/restapi.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/flask_rest_api/restapi.py b/utils/flask_rest_api/restapi.py index b93ad16a0f58..98cfa6dde305 100644 --- a/utils/flask_rest_api/restapi.py +++ b/utils/flask_rest_api/restapi.py @@ -32,6 +32,6 @@ def predict(): parser = argparse.ArgumentParser(description="Flask API exposing YOLOv5 model") parser.add_argument("--port", default=5000, type=int, help="port number") args = parser.parse_args() - + torch.hub._validate_not_a_forked_repo=lambda a,b,c: True model = torch.hub.load("ultralytics/yolov5", "yolov5s", force_reload=True) # force_reload to recache app.run(host="0.0.0.0", port=args.port) # debug=True causes Restarting with stat From 042874e06777e23eba84be715fb201945d0d60ec Mon Sep 17 00:00:00 2001 From: Zengyf-CVer <41098760+Zengyf-CVer@users.noreply.github.com> Date: Wed, 30 Mar 2022 16:20:01 +0800 Subject: [PATCH 2/5] Update restapi.py --- utils/flask_rest_api/restapi.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/flask_rest_api/restapi.py b/utils/flask_rest_api/restapi.py index 98cfa6dde305..3f99d1b44849 100644 --- a/utils/flask_rest_api/restapi.py +++ b/utils/flask_rest_api/restapi.py @@ -32,6 +32,6 @@ def predict(): parser = argparse.ArgumentParser(description="Flask API exposing YOLOv5 model") parser.add_argument("--port", default=5000, type=int, help="port number") args = parser.parse_args() - torch.hub._validate_not_a_forked_repo=lambda a,b,c: True + torch.hub._validate_not_a_forked_repo = lambda a,b,c: True model = torch.hub.load("ultralytics/yolov5", "yolov5s", force_reload=True) # force_reload to recache app.run(host="0.0.0.0", port=args.port) # debug=True causes Restarting with stat From 2af88c521b198cd6a0d2a467c70296848efdbd6e Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Sun, 3 Apr 2022 11:12:05 +0000 Subject: [PATCH 3/5] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- utils/flask_rest_api/restapi.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/flask_rest_api/restapi.py b/utils/flask_rest_api/restapi.py index 3f99d1b44849..9fa8d1b51645 100644 --- a/utils/flask_rest_api/restapi.py +++ b/utils/flask_rest_api/restapi.py @@ -32,6 +32,6 @@ def predict(): parser = argparse.ArgumentParser(description="Flask API exposing YOLOv5 model") parser.add_argument("--port", default=5000, type=int, help="port number") args = parser.parse_args() - torch.hub._validate_not_a_forked_repo = lambda a,b,c: True + torch.hub._validate_not_a_forked_repo = lambda a, b, c: True model = torch.hub.load("ultralytics/yolov5", "yolov5s", force_reload=True) # force_reload to recache app.run(host="0.0.0.0", port=args.port) # debug=True causes Restarting with stat From 155891321bb10e07f1fad2366b6275c3993630d5 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 3 Apr 2022 13:18:12 +0200 Subject: [PATCH 4/5] Cleanup --- utils/flask_rest_api/restapi.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/utils/flask_rest_api/restapi.py b/utils/flask_rest_api/restapi.py index 9fa8d1b51645..085884748fd2 100644 --- a/utils/flask_rest_api/restapi.py +++ b/utils/flask_rest_api/restapi.py @@ -1,12 +1,12 @@ """ -Run a rest API exposing the yolov5s object detection model +Run a Flask REST API exposing a YOLOv5s model """ import argparse import io import torch -from flask import Flask, request from PIL import Image +from flask import Flask, request app = Flask(__name__) @@ -31,7 +31,10 @@ def predict(): if __name__ == "__main__": parser = argparse.ArgumentParser(description="Flask API exposing YOLOv5 model") parser.add_argument("--port", default=5000, type=int, help="port number") - args = parser.parse_args() + opt = parser.parse_args() + + # Fix known issue urllib.error.HTTPError 403: rate limit exceeded https://github.com/ultralytics/yolov5/pull/7210 torch.hub._validate_not_a_forked_repo = lambda a, b, c: True + model = torch.hub.load("ultralytics/yolov5", "yolov5s", force_reload=True) # force_reload to recache - app.run(host="0.0.0.0", port=args.port) # debug=True causes Restarting with stat + app.run(host="0.0.0.0", port=opt.port) # debug=True causes Restarting with stat From ed9b986d7e3841469542a4e2ad6da87e7af05d6c Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Sun, 3 Apr 2022 11:18:28 +0000 Subject: [PATCH 5/5] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- utils/flask_rest_api/restapi.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/flask_rest_api/restapi.py b/utils/flask_rest_api/restapi.py index 085884748fd2..38868cc98d84 100644 --- a/utils/flask_rest_api/restapi.py +++ b/utils/flask_rest_api/restapi.py @@ -5,8 +5,8 @@ import io import torch -from PIL import Image from flask import Flask, request +from PIL import Image app = Flask(__name__)