From 9c13a349ce64982ff3d2f68578a9f78b88871a96 Mon Sep 17 00:00:00 2001
From: maxmon <541182180@qq.com>
Date: Fri, 18 Aug 2023 00:57:48 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E7=82=B9=E6=A0=87=E6=B3=A8=E6=8E=A5?=
 =?UTF-8?q?=E5=85=A5=20auto=5Fanno=20=E8=87=AA=E5=8A=A8=E6=A3=80=E6=B5=8B?=
 =?UTF-8?q?=E5=9F=BA=E7=A1=80=E7=9B=AE=E6=A0=87?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 fe/src/components/NER/index.vue | 66 +++++++++++++++++++++++++++++++++
 fe/yarn.lock                    |  6 +--
 2 files changed, 69 insertions(+), 3 deletions(-)

diff --git a/fe/src/components/NER/index.vue b/fe/src/components/NER/index.vue
index bf218ea..fb3a53b 100644
--- a/fe/src/components/NER/index.vue
+++ b/fe/src/components/NER/index.vue
@@ -25,6 +25,7 @@
       <div class="right">
         <div class="title">
           <span>选择标签：</span>
+          <button v-if="projectType === '图片点标注'" @click="autoAnno" class="auto-anno">自动标注</button>
           <div class="type-box">
             <span v-for="type in typeList" :key="type" @click="setType(type)" @contextmenu="delType(type, $event)" :class="isTypeSelected(type)?'type selected':'type'"
               :style="{
@@ -740,6 +741,48 @@ export default {
       } else {
         return this.nowType === type
       }
+    },
+    // 请求自动标注接口
+    autoAnno () {
+      const that = this
+      // 获取图片base64
+      const img = document.getElementById('anno-img')
+      const canvas = document.createElement('canvas')
+      const ctx = canvas.getContext('2d')
+      // 获取图片原始大小
+      const width = img.naturalWidth
+      const height = img.naturalHeight
+      // 设置canvas的宽高
+      canvas.width = width
+      canvas.height = height
+      // 绘制图片
+      ctx.drawImage(img, 0, 0, width, height)
+      // 获取base64
+      const base64 = canvas.toDataURL('image/png')
+      // 去掉前缀
+      const base64Data = base64.replace(/^data:image\/\w+;base64,/, '')
+      // 请求接口
+      const url = 'http://127.0.0.1:5000/detect'
+      post(url, {
+        projectName: that.projectName,
+        'image': base64Data
+      }, (info) => {
+        // 添加标注
+        const _ners = []
+        for (let i = 0; i < info.rect_angles.length; i++) {
+          const point = info.rect_angles[i]
+          const width = info.width
+          const height = info.height
+          console.log(point)
+          _ners.push({
+            'type': that.nowType,
+            'points': [[point.x / width, point.y / height]]
+          })
+        }
+        that.ners = [..._ners]
+        // 保存标注
+        that.save()
+      })
     }
   },
   watch: {
@@ -965,6 +1008,29 @@ export default {
   flex-direction: column;
   user-select: none;
 }
+.auto-anno {
+  float: right;
+  margin-right: 10px;
+  cursor: pointer;
+  box-sizing: content-box;
+  border: 2px solid #03022c;
+  text-align: center;
+  display: inline-block;
+  padding: 0 4px;
+  min-width: 100px;
+  height: 32px;
+  font-size: 18px;
+  line-height: 30px;
+  border-radius: 4px;
+  background: #03022c;
+  color: #fff;
+  font-weight: 500;
+  vertical-align: middle;
+  outline: none;
+}
+button:hover {
+  opacity: .7;
+}
 .type-box {
   display: inline;
   line-height: 40px;
diff --git a/fe/yarn.lock b/fe/yarn.lock
index c2fc358..88e2f3e 100644
--- a/fe/yarn.lock
+++ b/fe/yarn.lock
@@ -1856,10 +1856,10 @@ chownr@^1.0.1:
   resolved "https://registry.npm.taobao.org/chownr/download/chownr-1.1.4.tgz#6fc9d7b42d32a583596337666e7d08084da2cc6b"
   integrity sha1-b8nXtC0ypYNZYzdmbn0ICE2izGs=
 
-chromedriver@^2.27.2:
+chromedriver@^2.46.0:
   version "2.46.0"
-  resolved "https://registry.npm.taobao.org/chromedriver/download/chromedriver-2.46.0.tgz?cache=0&sync_timestamp=1614899323230&other_urls=https%3A%2F%2Fregistry.npm.taobao.org%2Fchromedriver%2Fdownload%2Fchromedriver-2.46.0.tgz#3d78e7eb9bb65dd804fe327a6bf76fced12be053"
-  integrity sha1-PXjn65u2XdgE/jJ6a/dvztEr4FM=
+  resolved "https://registry.npmmirror.com/chromedriver/-/chromedriver-2.46.0.tgz#3d78e7eb9bb65dd804fe327a6bf76fced12be053"
+  integrity sha512-dLtKIJW3y/PuFrPmcw6Mb8Nh+HwSqgVrK1rWgTARXhHfWvV822X2VRkx2meU/tg2+YQL6/nNgT6n5qWwIDHbwg==
   dependencies:
     del "^3.0.0"
     extract-zip "^1.6.7"