Skip to content

Commit cd2251c

Browse files
authored
update u2net-lite model link in background removal notebook (openvinotoolkit#1711)
1 parent 898d8d4 commit cd2251c

File tree

1 file changed

+6
-37
lines changed

1 file changed

+6
-37
lines changed

notebooks/205-vision-background-removal/205-vision-background-removal.ipynb

+6-37
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,8 @@
1414
"The PyTorch U$^2$-Net model is converted to OpenVINO IR format. The model source is available [here](https://github.com/xuebinqin/U-2-Net).\n",
1515
"\n",
1616
"\n",
17-
"#### Table of contents:\n\n",
17+
"#### Table of contents:\n",
18+
"\n",
1819
"- [Preparation](#Preparation)\n",
1920
" - [Install requirements](#Install-requirements)\n",
2021
" - [Import the PyTorch Library and U$^2$-Net](#Import-the-PyTorch-Library-and-U2-Net)\n",
@@ -150,7 +151,7 @@
150151
"\n",
151152
"u2net_lite = model_config(\n",
152153
" name=\"u2net_lite\",\n",
153-
" url=\"https://drive.google.com/uc?id=1rbSTGKAE-MTxBYHd-51l2hMOQPT_7EPy\",\n",
154+
" url=\"https://drive.google.com/uc?id=1W8E4FHIlTVstfRkYmNOjbr0VDXTZm0jD\",\n",
154155
" model=U2NETP,\n",
155156
" model_args=(),\n",
156157
")\n",
@@ -162,7 +163,7 @@
162163
")\n",
163164
"u2net_human_seg = model_config(\n",
164165
" name=\"u2net_human_seg\",\n",
165-
" url=\"https://drive.google.com/uc?id=1-Yg0cxgrNhHP-016FPdp902BR-kSsA4P\",\n",
166+
" url=\"https://drive.google.com/uc?id=1m_Kgs91b21gayc2XLW0ou8yugAIadWVP\",\n",
166167
" model=U2NET,\n",
167168
" model_args=(3, 1),\n",
168169
")\n",
@@ -701,44 +702,12 @@
701702
},
702703
"widgets": {
703704
"application/vnd.jupyter.widget-state+json": {
704-
"state": {
705-
"08172c8cd5e943cfb6a9e4e2ec33ab13": {
706-
"model_module": "@jupyter-widgets/controls",
707-
"model_module_version": "2.0.0",
708-
"model_name": "DropdownModel",
709-
"state": {
710-
"_options_labels": [
711-
"CPU",
712-
"GPU.0",
713-
"GPU.1",
714-
"AUTO"
715-
],
716-
"description": "Device:",
717-
"index": 3,
718-
"layout": "IPY_MODEL_61f88e29badf43fe9ebabb3ecbc56e2a",
719-
"style": "IPY_MODEL_b186e86e45b94c2ca7c09c2933ace135"
720-
}
721-
},
722-
"61f88e29badf43fe9ebabb3ecbc56e2a": {
723-
"model_module": "@jupyter-widgets/base",
724-
"model_module_version": "2.0.0",
725-
"model_name": "LayoutModel",
726-
"state": {}
727-
},
728-
"b186e86e45b94c2ca7c09c2933ace135": {
729-
"model_module": "@jupyter-widgets/controls",
730-
"model_module_version": "2.0.0",
731-
"model_name": "DescriptionStyleModel",
732-
"state": {
733-
"description_width": ""
734-
}
735-
}
736-
},
705+
"state": {},
737706
"version_major": 2,
738707
"version_minor": 0
739708
}
740709
}
741710
},
742711
"nbformat": 4,
743712
"nbformat_minor": 5
744-
}
713+
}

0 commit comments

Comments
 (0)