From 54d3908a0cefbb01b3eb1592b1d54afd452d9e21 Mon Sep 17 00:00:00 2001
From: "Kin-Yiu, Wong" <102582011@cc.ncu.edu.tw>
Date: Tue, 9 Aug 2022 15:23:42 +0800
Subject: [PATCH] main code

update anchors
---
 tools/reparameterization.ipynb | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/tools/reparameterization.ipynb b/tools/reparameterization.ipynb
index 84e43269..d529884b 100644
--- a/tools/reparameterization.ipynb
+++ b/tools/reparameterization.ipynb
@@ -38,7 +38,7 @@
     "\n",
     "with open('cfg/deploy/yolov7.yaml') as f:\n",
     "    yml = yaml.load(f, Loader=yaml.SafeLoader)\n",
-    "anchors = len(yml['anchors'])\n",
+    "anchors = len(yml['anchors'][0]) // 2\n",
     "\n",
     "# copy intersect weights\n",
     "state_dict = ckpt['model'].float().state_dict()\n",
@@ -100,7 +100,7 @@
     "\n",
     "with open('cfg/deploy/yolov7x.yaml') as f:\n",
     "    yml = yaml.load(f, Loader=yaml.SafeLoader)\n",
-    "anchors = len(yml['anchors'])\n",
+    "anchors = len(yml['anchors'][0]) // 2\n",
     "\n",
     "# copy intersect weights\n",
     "state_dict = ckpt['model'].float().state_dict()\n",
@@ -162,7 +162,7 @@
     "\n",
     "with open('cfg/deploy/yolov7-w6.yaml') as f:\n",
     "    yml = yaml.load(f, Loader=yaml.SafeLoader)\n",
-    "anchors = len(yml['anchors'])\n",
+    "anchors = len(yml['anchors'][0]) // 2\n",
     "\n",
     "# copy intersect weights\n",
     "state_dict = ckpt['model'].float().state_dict()\n",
@@ -248,7 +248,7 @@
     "\n",
     "with open('cfg/deploy/yolov7-e6.yaml') as f:\n",
     "    yml = yaml.load(f, Loader=yaml.SafeLoader)\n",
-    "anchors = len(yml['anchors'])\n",
+    "anchors = len(yml['anchors'][0]) // 2\n",
     "\n",
     "# copy intersect weights\n",
     "state_dict = ckpt['model'].float().state_dict()\n",
@@ -334,7 +334,7 @@
     "\n",
     "with open('cfg/deploy/yolov7-d6.yaml') as f:\n",
     "    yml = yaml.load(f, Loader=yaml.SafeLoader)\n",
-    "anchors = len(yml['anchors'])\n",
+    "anchors = len(yml['anchors'][0]) // 2\n",
     "\n",
     "# copy intersect weights\n",
     "state_dict = ckpt['model'].float().state_dict()\n",
@@ -420,7 +420,7 @@
     "\n",
     "with open('cfg/deploy/yolov7-e6e.yaml') as f:\n",
     "    yml = yaml.load(f, Loader=yaml.SafeLoader)\n",
-    "anchors = len(yml['anchors'])\n",
+    "anchors = len(yml['anchors'][0]) // 2)\n",
     "\n",
     "# copy intersect weights\n",
     "state_dict = ckpt['model'].float().state_dict()\n",