add luojianet

app.py

@@ -0,0 +1,102 @@
+import os
+import requests
+import json
+import time
+import gradio as gr
+
+from utils import get_token
+from obshandler import OBSHandler
+
+url = os.environ["URL_NODE"]
+
+obs = OBSHandler()
+
+
+def detect_image(image):
+    print("image: ", image)
+
+    user_name = "huggingface"
+    upload_path = user_name + "/" + str(time.time()) + "/input.jpg"
+    obs.upload_file(upload_path, image)
+
+    token = get_token()
+
+    requests_json = {"file_path": upload_path}
+    headers = {"Content-Type": "application/json", "X-Auth-Token": token}
+    resp = requests.post(url,
+                         json=requests_json,
+                         headers=headers,
+                         verify=False)
+    resp = json.loads(resp.text)
+    gen_url = resp["result"]
+    return gen_url
+
+
+def read_content(file_path):
+    with open(file_path, 'r', encoding='utf-8') as f:
+        content = f.read()
+    return content
+
+
+example_images = [
+    os.path.join(os.path.dirname(__file__), "examples/00.jpg"),
+    os.path.join(os.path.dirname(__file__), "examples/01.jpg"),
+    os.path.join(os.path.dirname(__file__), "examples/02.jpg"),
+    os.path.join(os.path.dirname(__file__), "examples/03.jpg"),
+    os.path.join(os.path.dirname(__file__), "examples/04.jpg"),
+    os.path.join(os.path.dirname(__file__), "examples/05.jpg")
+]
+
+default_image = example_images[0]
+
+css = """
+.gradio-container {background-image: url('file=./background.jpg'); background-size:cover; background-repeat: no-repeat;}
+"""
+
+# warm up
+# detect_image()
+
+with gr.Blocks(css=css) as demo:
+    gr.HTML(read_content("./header.html"))
+    gr.Markdown("# MindSpore Wuhan.LuoJiaNET")
+    gr.Markdown(
+        "`Wuhan.LuoJiaNET` is the first domestic autonomous and controllable machine learning framework for remote sensing in the field of remote sensing,"
+        " jointly developed by` Wuhan University` and `Huawei's Ascend AI team`, which has the characteristics of large image size,"
+        " multiple data channels, and large scale variation of remote sensing data."
+        " It is compatible with existing deep learning frameworks and provides a user-friendly,"
+        " drag-and-drop interactive network structure to build an interface."
+        " It can shield the differences between different hardware devices and manage a diversified remote sensing image sample library,"
+        " LuoJiaSET, to achieve efficient storage and management of remote multi-source sensing image samples."
+    )
+
+    with gr.Tab("目标识别 (Object Detection)"):
+        with gr.Row():
+            image_input = gr.Image(type="filepath",
+                                   value=default_image,
+                                   interactive=False)
+            image_output = gr.Image(type="filepath")
+
+        gr.Examples(
+            examples=example_images,
+            inputs=image_input,
+        )
+        image_button = gr.Button("Detect")
+
+    with gr.Accordion("Open for More!"):
+        gr.Markdown(
+            "- If you want to know more about the foundation models of MindSpore, please visit "
+            "[The Foundation Models Platform for Mindspore](https://xihe.mindspore.cn/)"
+        )
+        gr.Markdown(
+            "- If you want to know more about Wuhan.LuoJiaNET, please visit "
+            "[Wuhan.LuoJiaNET](https://github.com/WHULuoJiaTeam/luojianet)")
+        gr.Markdown(
+            "- Try [Wukong-LuojiaNET model on the Foundation Models Platform for Mindspore]"
+            "(https://xihe.mindspore.cn/modelzoo/luojia)")
+
+    image_button.click(detect_image,
+                       inputs=[image_input],
+                       outputs=[image_output])
+
+demo.queue(concurrency_count=5)
+demo.launch(enable_queue=True)

header.html

@@ -0,0 +1,27 @@
+<div style="text-align: center; max-width: 1920px; margin: 0 auto;">
+    <div
+      style="
+        display: inline-flex;
+        gap: 0.8rem;
+        font-size: 1.75rem;
+        margin-bottom: 10px;
+        margin-left: 220px;
+        justify-content: center;
+      "
+    >
+    </div>
+    <div
+      style="
+        display: inline-flex;
+        align-items: center;
+        gap: 0.8rem;
+        font-size: 1.75rem;
+        margin-bottom: 10px;
+        justify-content: center;
+      ">
+    <a href="https://github.com/mindspore-ai/mindspore"><h1 style="font-weight: 900; align-items: center; margin-bottom: 7px;">
+    </h1></a>
+    </div>
+    <a href="https://github.com/mindspore-ai/mindspore"><img src="https://xihe.mindspore.cn/assets/modelzoo2.3fcf3c59.jpg" width="100%"></a>
+
+  </div>

obshandler.py

@@ -0,0 +1,54 @@
+import os
+import traceback
+
+from obs import ObsClient
+from obs import PutObjectHeader
+
+AK = os.environ["OBS_AK"]
+SK = os.environ["OBS_SK"]
+
+ENDPOINT = os.environ["OBS_ENDPOINT"]
+BUCKET_NAME = os.environ["BUCKET"]
+
+
+class OBSHandler:
+    def __init__(self):
+        self.access_key = AK
+        self.secret_key = SK
+        self.endpoint = ENDPOINT
+        self.bucket_name = BUCKET_NAME
+        self.server = "https://" + self.endpoint
+        self.obsClient = self.init_obs()
+        self.maxkeys = 1000  # 查询的对象最大个数
+
+    # 初始化obs
+    def init_obs(self):
+        obsClient = ObsClient(access_key_id=self.access_key,
+                              secret_access_key=self.secret_key,
+                              server=self.server)
+        return obsClient
+
+    def close_obs(self):
+        self.obsClient.close()
+
+    def upload_file(self, dest, source):
+        response_msg = {'status': 200, 'msg': '单个对象上传成功'}
+        headers = PutObjectHeader()
+        headers.contentType = 'image/jpeg'
+        try:
+            resp = self.obsClient.putFile(self.bucket_name,
+                                          dest,
+                                          source,
+                                          headers=headers)
+
+            if resp.status < 300:
+                print("Upload image to obs successfully!")
+            else:
+                print('errorCode:', resp.errorCode)
+                print('errorMessage:', resp.errorMessage)
+        except Exception:
+            print(traceback.format_exc())
+            response_msg["status"] = -1
+            response_msg["msg"] = "单个对象上传失败"
+            print("Upload image to obs error!")
+        return response_msg

utils.py

@@ -0,0 +1,46 @@
+import os
+import requests
+
+
+def get_token():
+    username = os.environ["USER_NAME"]
+    domain_name = os.environ["DOMAIN_NAME"]
+    domain_pwd = os.environ["DOMAIN_PWD"]
+    url = os.environ["IAM_URL"]
+
+    requests_json = {
+        "auth": {
+            "identity": {
+                "methods": ["password"],
+                "password": {
+                    "user": {
+                        "name": username,
+                        "password": domain_pwd,
+                        "domain": {
+                            "name": domain_name
+                        }
+                    }
+                }
+            },
+            "scope": {
+                "project": {
+                    "name": "cn-central-221"
+                }
+            }
+        }
+    }
+
+    headers = {
+        "Content-Type": "application/json"
+    }
+
+    response = requests.post(url, json=requests_json, headers=headers)
+
+    result = response.headers
+    print("token success")
+
+    return result['X-Subject-Token']
+
+
+if __name__ == "__main__":
+    get_token()