apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

main code

Browse Source 
update anchors
This commit is contained in:
Kin-Yiu, Wong 2022-08-09 15:23:42 +08:00 committed by GitHub
parent fef71ef42c
commit 54d3908a0c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tools/reparameterization.ipynb
View File

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