14
14
"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",
15
15
"\n",
16
16
"\n",
17
- "#### Table of contents:\n\n",
17
+ "#### Table of contents:\n",
18
+ "\n",
18
19
"- [Preparation](#Preparation)\n",
19
20
" - [Install requirements](#Install-requirements)\n",
20
21
" - [Import the PyTorch Library and U$^2$-Net](#Import-the-PyTorch-Library-and-U2-Net)\n",
150
151
"\n",
151
152
"u2net_lite = model_config(\n",
152
153
" 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",
154
155
" model=U2NETP,\n",
155
156
" model_args=(),\n",
156
157
")\n",
162
163
")\n",
163
164
"u2net_human_seg = model_config(\n",
164
165
" 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",
166
167
" model=U2NET,\n",
167
168
" model_args=(3, 1),\n",
168
169
")\n",
701
702
},
702
703
"widgets": {
703
704
"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": {},
737
706
"version_major": 2,
738
707
"version_minor": 0
739
708
}
740
709
}
741
710
},
742
711
"nbformat": 4,
743
712
"nbformat_minor": 5
744
- }
713
+ }
0 commit comments