mirror of
https://github.com/mmaithani/data-science.git
synced 2022-04-24 02:56:41 +03:00
Created using Colaboratory
This commit is contained in:
Mohit Maithani
338
OneNet.ipynb
338
OneNet.ipynb
@@ -7,7 +7,6 @@
|
|||||||
"name": "OneNet.ipynb",
|
"name": "OneNet.ipynb",
|
||||||
"provenance": [],
|
"provenance": [],
|
||||||
"collapsed_sections": [],
|
"collapsed_sections": [],
|
||||||
"toc_visible": true,
|
|
||||||
"machine_shape": "hm",
|
"machine_shape": "hm",
|
||||||
"include_colab_link": true
|
"include_colab_link": true
|
||||||
},
|
},
|
||||||
@@ -56,62 +55,6 @@
|
|||||||
}
|
}
|
||||||
]
|
]
|
||||||
},
|
},
|
||||||
{
|
|
||||||
"cell_type": "markdown",
|
|
||||||
"metadata": {
|
|
||||||
"id": "GYf_YDasiXXx"
|
|
||||||
},
|
|
||||||
"source": [
|
|
||||||
"#download the pytorch version suitable with your cuda version"
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"cell_type": "code",
|
|
||||||
"metadata": {
|
|
||||||
"colab": {
|
|
||||||
"base_uri": "https://localhost:8080/"
|
|
||||||
},
|
|
||||||
"id": "3MxmM0p2ZBAL",
|
|
||||||
"outputId": "d4c2d864-1487-4860-9f36-7efd3df10c7a"
|
|
||||||
},
|
|
||||||
"source": [
|
|
||||||
"!pip install torch===1.7.1+cu110 torchvision===0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html"
|
|
||||||
],
|
|
||||||
"execution_count": 2,
|
|
||||||
"outputs": [
|
|
||||||
{
|
|
||||||
"output_type": "stream",
|
|
||||||
"text": [
|
|
||||||
"Looking in links: https://download.pytorch.org/whl/torch_stable.html\n",
|
|
||||||
"Collecting torch===1.7.1+cu110\n",
|
|
||||||
"\u001b[?25l Downloading https://download.pytorch.org/whl/cu110/torch-1.7.1%2Bcu110-cp36-cp36m-linux_x86_64.whl (1156.8MB)\n",
|
|
||||||
"\u001b[K |███████████████████████ | 834.1MB 1.3MB/s eta 0:04:07tcmalloc: large alloc 1147494400 bytes == 0x39552000 @ 0x7f2169075615 0x591e47 0x4cc179 0x4cc2db 0x50a1cc 0x50beb4 0x507be4 0x509900 0x50a2fd 0x50beb4 0x507be4 0x509900 0x50a2fd 0x50cc96 0x58e683 0x50c127 0x58e683 0x50c127 0x58e683 0x50c127 0x58e683 0x5f7832 0x4e5493 0x5515c1 0x5a9dac 0x50a433 0x50cc96 0x5095c8 0x50a2fd 0x50beb4 0x507be4\n",
|
|
||||||
"\u001b[K |█████████████████████████████▏ | 1055.7MB 1.2MB/s eta 0:01:24tcmalloc: large alloc 1434370048 bytes == 0x7dba8000 @ 0x7f2169075615 0x591e47 0x4cc179 0x4cc2db 0x50a1cc 0x50beb4 0x507be4 0x509900 0x50a2fd 0x50beb4 0x507be4 0x509900 0x50a2fd 0x50cc96 0x58e683 0x50c127 0x58e683 0x50c127 0x58e683 0x50c127 0x58e683 0x5f7832 0x4e5493 0x5515c1 0x5a9dac 0x50a433 0x50cc96 0x5095c8 0x50a2fd 0x50beb4 0x507be4\n",
|
|
||||||
"\u001b[K |████████████████████████████████| 1156.8MB 1.2MB/s eta 0:00:01tcmalloc: large alloc 1445953536 bytes == 0xd3394000 @ 0x7f2169075615 0x591e47 0x4cc179 0x4cc2db 0x50a1cc 0x50beb4 0x507be4 0x509900 0x50a2fd 0x50beb4 0x507be4 0x509900 0x50a2fd 0x50beb4 0x507be4 0x509900 0x50a2fd 0x50beb4 0x507be4 0x509900 0x50a2fd 0x50beb4 0x507be4 0x509900 0x50a2fd 0x50beb4 0x5095c8 0x50a2fd 0x50beb4 0x507be4 0x509900\n",
|
|
||||||
"\u001b[K |████████████████████████████████| 1156.8MB 15kB/s \n",
|
|
||||||
"\u001b[?25hCollecting torchvision===0.8.2+cu110\n",
|
|
||||||
"\u001b[?25l Downloading https://download.pytorch.org/whl/cu110/torchvision-0.8.2%2Bcu110-cp36-cp36m-linux_x86_64.whl (12.9MB)\n",
|
|
||||||
"\u001b[K |████████████████████████████████| 12.9MB 101kB/s \n",
|
|
||||||
"\u001b[?25hCollecting torchaudio===0.7.2\n",
|
|
||||||
"\u001b[?25l Downloading https://files.pythonhosted.org/packages/2a/f9/618434cf4e46dc975871e1516f5499abef6564ab4366f9b2321ee536be14/torchaudio-0.7.2-cp36-cp36m-manylinux1_x86_64.whl (7.6MB)\n",
|
|
||||||
"\u001b[K |████████████████████████████████| 7.6MB 3.2MB/s \n",
|
|
||||||
"\u001b[?25hRequirement already satisfied: numpy in /usr/local/lib/python3.6/dist-packages (from torch===1.7.1+cu110) (1.18.5)\n",
|
|
||||||
"Requirement already satisfied: dataclasses; python_version < \"3.7\" in /usr/local/lib/python3.6/dist-packages (from torch===1.7.1+cu110) (0.8)\n",
|
|
||||||
"Requirement already satisfied: typing-extensions in /usr/local/lib/python3.6/dist-packages (from torch===1.7.1+cu110) (3.7.4.3)\n",
|
|
||||||
"Requirement already satisfied: pillow>=4.1.1 in /usr/local/lib/python3.6/dist-packages (from torchvision===0.8.2+cu110) (7.0.0)\n",
|
|
||||||
"Installing collected packages: torch, torchvision, torchaudio\n",
|
|
||||||
" Found existing installation: torch 1.7.0+cu101\n",
|
|
||||||
" Uninstalling torch-1.7.0+cu101:\n",
|
|
||||||
" Successfully uninstalled torch-1.7.0+cu101\n",
|
|
||||||
" Found existing installation: torchvision 0.8.1+cu101\n",
|
|
||||||
" Uninstalling torchvision-0.8.1+cu101:\n",
|
|
||||||
" Successfully uninstalled torchvision-0.8.1+cu101\n",
|
|
||||||
"Successfully installed torch-1.7.1+cu110 torchaudio-0.7.2 torchvision-0.8.2+cu110\n"
|
|
||||||
],
|
|
||||||
"name": "stdout"
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
{
|
||||||
"cell_type": "code",
|
"cell_type": "code",
|
||||||
"metadata": {
|
"metadata": {
|
||||||
@@ -135,6 +78,100 @@
|
|||||||
}
|
}
|
||||||
]
|
]
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
"cell_type": "markdown",
|
||||||
|
"metadata": {
|
||||||
|
"id": "CxGYhfl3rCy4"
|
||||||
|
},
|
||||||
|
"source": [
|
||||||
|
"# installing torchvison according to your cuda version"
|
||||||
|
]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"cell_type": "code",
|
||||||
|
"metadata": {
|
||||||
|
"colab": {
|
||||||
|
"base_uri": "https://localhost:8080/"
|
||||||
|
},
|
||||||
|
"id": "ora4M9KThIwk",
|
||||||
|
"outputId": "9308617f-3086-451a-906a-188577256692"
|
||||||
|
},
|
||||||
|
"source": [
|
||||||
|
"!nvcc --version"
|
||||||
|
],
|
||||||
|
"execution_count": 9,
|
||||||
|
"outputs": [
|
||||||
|
{
|
||||||
|
"output_type": "stream",
|
||||||
|
"text": [
|
||||||
|
"nvcc: NVIDIA (R) Cuda compiler driver\n",
|
||||||
|
"Copyright (c) 2005-2019 NVIDIA Corporation\n",
|
||||||
|
"Built on Sun_Jul_28_19:07:16_PDT_2019\n",
|
||||||
|
"Cuda compilation tools, release 10.1, V10.1.243\n"
|
||||||
|
],
|
||||||
|
"name": "stdout"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"cell_type": "code",
|
||||||
|
"metadata": {
|
||||||
|
"colab": {
|
||||||
|
"base_uri": "https://localhost:8080/",
|
||||||
|
"height": 462
|
||||||
|
},
|
||||||
|
"id": "aLOtDDDihcy0",
|
||||||
|
"outputId": "14d35671-6a99-49e5-afff-cbd95fd37e9e"
|
||||||
|
},
|
||||||
|
"source": [
|
||||||
|
"pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html"
|
||||||
|
],
|
||||||
|
"execution_count": 10,
|
||||||
|
"outputs": [
|
||||||
|
{
|
||||||
|
"output_type": "stream",
|
||||||
|
"text": [
|
||||||
|
"Looking in links: https://download.pytorch.org/whl/torch_stable.html\n",
|
||||||
|
"Collecting torch==1.7.1+cu101\n",
|
||||||
|
"\u001b[?25l Downloading https://download.pytorch.org/whl/cu101/torch-1.7.1%2Bcu101-cp36-cp36m-linux_x86_64.whl (735.4MB)\n",
|
||||||
|
"\u001b[K |████████████████████████████████| 735.4MB 23kB/s \n",
|
||||||
|
"\u001b[?25hCollecting torchvision==0.8.2+cu101\n",
|
||||||
|
"\u001b[?25l Downloading https://download.pytorch.org/whl/cu101/torchvision-0.8.2%2Bcu101-cp36-cp36m-linux_x86_64.whl (12.8MB)\n",
|
||||||
|
"\u001b[K |████████████████████████████████| 12.8MB 14kB/s \n",
|
||||||
|
"\u001b[?25hRequirement already satisfied: torchaudio===0.7.2 in /usr/local/lib/python3.6/dist-packages (0.7.2)\n",
|
||||||
|
"Requirement already satisfied: dataclasses; python_version < \"3.7\" in /usr/local/lib/python3.6/dist-packages (from torch==1.7.1+cu101) (0.8)\n",
|
||||||
|
"Requirement already satisfied: typing-extensions in /usr/local/lib/python3.6/dist-packages (from torch==1.7.1+cu101) (3.7.4.3)\n",
|
||||||
|
"Requirement already satisfied: numpy in /usr/local/lib/python3.6/dist-packages (from torch==1.7.1+cu101) (1.18.5)\n",
|
||||||
|
"Requirement already satisfied: pillow>=4.1.1 in /usr/local/lib/python3.6/dist-packages (from torchvision==0.8.2+cu101) (7.0.0)\n",
|
||||||
|
"Installing collected packages: torch, torchvision\n",
|
||||||
|
" Found existing installation: torch 1.7.1+cu110\n",
|
||||||
|
" Uninstalling torch-1.7.1+cu110:\n",
|
||||||
|
" Successfully uninstalled torch-1.7.1+cu110\n",
|
||||||
|
" Found existing installation: torchvision 0.8.2+cu110\n",
|
||||||
|
" Uninstalling torchvision-0.8.2+cu110:\n",
|
||||||
|
" Successfully uninstalled torchvision-0.8.2+cu110\n",
|
||||||
|
"Successfully installed torch-1.7.1+cu101 torchvision-0.8.2+cu101\n"
|
||||||
|
],
|
||||||
|
"name": "stdout"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"output_type": "display_data",
|
||||||
|
"data": {
|
||||||
|
"application/vnd.colab-display-data+json": {
|
||||||
|
"pip_warning": {
|
||||||
|
"packages": [
|
||||||
|
"torch",
|
||||||
|
"torchvision"
|
||||||
|
]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"metadata": {
|
||||||
|
"tags": []
|
||||||
|
}
|
||||||
|
}
|
||||||
|
]
|
||||||
|
},
|
||||||
{
|
{
|
||||||
"cell_type": "markdown",
|
"cell_type": "markdown",
|
||||||
"metadata": {
|
"metadata": {
|
||||||
@@ -158,34 +195,14 @@
|
|||||||
{
|
{
|
||||||
"cell_type": "code",
|
"cell_type": "code",
|
||||||
"metadata": {
|
"metadata": {
|
||||||
"colab": {
|
"id": "f5-hLPy_Zg9p"
|
||||||
"base_uri": "https://localhost:8080/"
|
|
||||||
},
|
|
||||||
"id": "f5-hLPy_Zg9p",
|
|
||||||
"outputId": "400e7c83-9511-4333-e118-7a3dd68c32ab"
|
|
||||||
},
|
},
|
||||||
"source": [
|
"source": [
|
||||||
"import torch, torchvision\r\n",
|
"import torch, torchvision\r\n",
|
||||||
"print(torch.__version__, torch.cuda.is_available())"
|
"print(torch.__version__, torch.cuda.is_available())"
|
||||||
],
|
],
|
||||||
"execution_count": 4,
|
"execution_count": null,
|
||||||
"outputs": [
|
"outputs": []
|
||||||
{
|
|
||||||
"output_type": "stream",
|
|
||||||
"text": [
|
|
||||||
"1.7.1+cu110 False\n"
|
|
||||||
],
|
|
||||||
"name": "stdout"
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"output_type": "stream",
|
|
||||||
"text": [
|
|
||||||
"/usr/local/lib/python3.6/dist-packages/torch/cuda/__init__.py:52: UserWarning: CUDA initialization: The NVIDIA driver on your system is too old (found version 10010). Please update your GPU driver by downloading and installing a new version from the URL: http://www.nvidia.com/Download/index.aspx Alternatively, go to: https://pytorch.org to install a PyTorch version that has been compiled with your version of the CUDA driver. (Triggered internally at /pytorch/c10/cuda/CUDAFunctions.cpp:100.)\n",
|
|
||||||
" return torch._C._cuda_getDeviceCount() > 0\n"
|
|
||||||
],
|
|
||||||
"name": "stderr"
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
"cell_type": "markdown",
|
"cell_type": "markdown",
|
||||||
@@ -814,14 +831,20 @@
|
|||||||
}
|
}
|
||||||
]
|
]
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
"cell_type": "markdown",
|
||||||
|
"metadata": {
|
||||||
|
"id": "EZ4aOsnUqdeL"
|
||||||
|
},
|
||||||
|
"source": [
|
||||||
|
"# link coco datset to OneNet dateset folder\r\n",
|
||||||
|
"## download COCO dataset and then link the folder with OneNet"
|
||||||
|
]
|
||||||
|
},
|
||||||
{
|
{
|
||||||
"cell_type": "code",
|
"cell_type": "code",
|
||||||
"metadata": {
|
"metadata": {
|
||||||
"colab": {
|
"id": "6TNWAF7bW_bN"
|
||||||
"base_uri": "https://localhost:8080/"
|
|
||||||
},
|
|
||||||
"id": "6TNWAF7bW_bN",
|
|
||||||
"outputId": "27738eba-959f-4fa2-cf56-a676583fe644"
|
|
||||||
},
|
},
|
||||||
"source": [
|
"source": [
|
||||||
"!mkdir -p datasets/coco\r\n",
|
"!mkdir -p datasets/coco\r\n",
|
||||||
@@ -829,18 +852,8 @@
|
|||||||
"!ln -s /path_to_coco_dataset/train2017 datasets/coco/train2017\r\n",
|
"!ln -s /path_to_coco_dataset/train2017 datasets/coco/train2017\r\n",
|
||||||
"!ln -s /path_to_coco_dataset/val2017 datasets/coco/val2017"
|
"!ln -s /path_to_coco_dataset/val2017 datasets/coco/val2017"
|
||||||
],
|
],
|
||||||
"execution_count": 3,
|
"execution_count": 8,
|
||||||
"outputs": [
|
"outputs": []
|
||||||
{
|
|
||||||
"output_type": "stream",
|
|
||||||
"text": [
|
|
||||||
"ln: failed to create symbolic link 'datasets/coco/annotations': File exists\n",
|
|
||||||
"ln: failed to create symbolic link 'datasets/coco/train2017': File exists\n",
|
|
||||||
"ln: failed to create symbolic link 'datasets/coco/val2017': File exists\n"
|
|
||||||
],
|
|
||||||
"name": "stdout"
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
"cell_type": "code",
|
"cell_type": "code",
|
||||||
@@ -877,88 +890,12 @@
|
|||||||
]
|
]
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
"cell_type": "code",
|
"cell_type": "markdown",
|
||||||
"metadata": {
|
"metadata": {
|
||||||
"colab": {
|
"id": "qQOQZKS5rI3J"
|
||||||
"base_uri": "https://localhost:8080/"
|
|
||||||
},
|
|
||||||
"id": "ora4M9KThIwk",
|
|
||||||
"outputId": "90bd9303-c0ea-4cb1-a595-e588d1a5e538"
|
|
||||||
},
|
},
|
||||||
"source": [
|
"source": [
|
||||||
"!nvcc --version"
|
"# train ONeNet"
|
||||||
],
|
|
||||||
"execution_count": 7,
|
|
||||||
"outputs": [
|
|
||||||
{
|
|
||||||
"output_type": "stream",
|
|
||||||
"text": [
|
|
||||||
"nvcc: NVIDIA (R) Cuda compiler driver\n",
|
|
||||||
"Copyright (c) 2005-2019 NVIDIA Corporation\n",
|
|
||||||
"Built on Sun_Jul_28_19:07:16_PDT_2019\n",
|
|
||||||
"Cuda compilation tools, release 10.1, V10.1.243\n"
|
|
||||||
],
|
|
||||||
"name": "stdout"
|
|
||||||
}
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"cell_type": "code",
|
|
||||||
"metadata": {
|
|
||||||
"colab": {
|
|
||||||
"base_uri": "https://localhost:8080/",
|
|
||||||
"height": 442
|
|
||||||
},
|
|
||||||
"id": "aLOtDDDihcy0",
|
|
||||||
"outputId": "f02b26a5-b392-4154-bbc6-93f83ba38d19"
|
|
||||||
},
|
|
||||||
"source": [
|
|
||||||
"pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html"
|
|
||||||
],
|
|
||||||
"execution_count": 11,
|
|
||||||
"outputs": [
|
|
||||||
{
|
|
||||||
"output_type": "stream",
|
|
||||||
"text": [
|
|
||||||
"Looking in links: https://download.pytorch.org/whl/torch_stable.html\n",
|
|
||||||
"Collecting torch==1.7.1+cu101\n",
|
|
||||||
"\u001b[?25l Downloading https://download.pytorch.org/whl/cu101/torch-1.7.1%2Bcu101-cp36-cp36m-linux_x86_64.whl (735.4MB)\n",
|
|
||||||
"\u001b[K |████████████████████████████████| 735.4MB 22kB/s \n",
|
|
||||||
"\u001b[?25hCollecting torchvision==0.8.2+cu101\n",
|
|
||||||
"\u001b[?25l Downloading https://download.pytorch.org/whl/cu101/torchvision-0.8.2%2Bcu101-cp36-cp36m-linux_x86_64.whl (12.8MB)\n",
|
|
||||||
"\u001b[K |████████████████████████████████| 12.8MB 158kB/s \n",
|
|
||||||
"\u001b[?25hRequirement already satisfied: torchaudio===0.7.2 in /usr/local/lib/python3.6/dist-packages (0.7.2)\n",
|
|
||||||
"Requirement already satisfied: typing-extensions in /usr/local/lib/python3.6/dist-packages (from torch==1.7.1+cu101) (3.7.4.3)\n",
|
|
||||||
"Requirement already satisfied: dataclasses; python_version < \"3.7\" in /usr/local/lib/python3.6/dist-packages (from torch==1.7.1+cu101) (0.8)\n",
|
|
||||||
"Requirement already satisfied: numpy in /usr/local/lib/python3.6/dist-packages (from torch==1.7.1+cu101) (1.18.5)\n",
|
|
||||||
"Requirement already satisfied: pillow>=4.1.1 in /usr/local/lib/python3.6/dist-packages (from torchvision==0.8.2+cu101) (7.0.0)\n",
|
|
||||||
"Installing collected packages: torch, torchvision\n",
|
|
||||||
" Found existing installation: torch 1.7.1+cu110\n",
|
|
||||||
" Uninstalling torch-1.7.1+cu110:\n",
|
|
||||||
" Successfully uninstalled torch-1.7.1+cu110\n",
|
|
||||||
" Found existing installation: torchvision 0.8.2+cu110\n",
|
|
||||||
" Uninstalling torchvision-0.8.2+cu110:\n",
|
|
||||||
" Successfully uninstalled torchvision-0.8.2+cu110\n",
|
|
||||||
"Successfully installed torch-1.7.1+cu101 torchvision-0.8.2+cu101\n"
|
|
||||||
],
|
|
||||||
"name": "stdout"
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"output_type": "display_data",
|
|
||||||
"data": {
|
|
||||||
"application/vnd.colab-display-data+json": {
|
|
||||||
"pip_warning": {
|
|
||||||
"packages": [
|
|
||||||
"torch",
|
|
||||||
"torchvision"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"metadata": {
|
|
||||||
"tags": []
|
|
||||||
}
|
|
||||||
}
|
|
||||||
]
|
]
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
@@ -1613,40 +1550,12 @@
|
|||||||
]
|
]
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
"cell_type": "code",
|
"cell_type": "markdown",
|
||||||
"metadata": {
|
"metadata": {
|
||||||
"colab": {
|
"id": "FEQFguZWrMRm"
|
||||||
"base_uri": "https://localhost:8080/"
|
|
||||||
},
|
|
||||||
"id": "2DiMDFpYgoE4",
|
|
||||||
"outputId": "304233be-5b83-4583-b4c3-35f5cb933771"
|
|
||||||
},
|
},
|
||||||
"source": [
|
"source": [
|
||||||
"!pip install -U 'git+https://github.com/facebookresearch/iopath'"
|
"# Evaluate"
|
||||||
],
|
|
||||||
"execution_count": null,
|
|
||||||
"outputs": [
|
|
||||||
{
|
|
||||||
"output_type": "stream",
|
|
||||||
"text": [
|
|
||||||
"Collecting git+https://github.com/facebookresearch/iopath\n",
|
|
||||||
" Cloning https://github.com/facebookresearch/iopath to /tmp/pip-req-build-gttv2hv0\n",
|
|
||||||
" Running command git clone -q https://github.com/facebookresearch/iopath /tmp/pip-req-build-gttv2hv0\n",
|
|
||||||
"Requirement already satisfied, skipping upgrade: tqdm in /usr/local/lib/python3.6/dist-packages (from iopath==0.1.2) (4.41.1)\n",
|
|
||||||
"Requirement already satisfied, skipping upgrade: portalocker in /usr/local/lib/python3.6/dist-packages/portalocker-2.0.0-py3.6.egg (from iopath==0.1.2) (2.0.0)\n",
|
|
||||||
"Building wheels for collected packages: iopath\n",
|
|
||||||
" Building wheel for iopath (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
|
|
||||||
" Created wheel for iopath: filename=iopath-0.1.2-cp36-none-any.whl size=11409 sha256=a7e1d36e88dc291e8f47cc0f2d2078c0534ead53853973a606207e0c90b42591\n",
|
|
||||||
" Stored in directory: /tmp/pip-ephem-wheel-cache-mh1c79pt/wheels/c3/67/ba/08e69262171ec0da0fd614e8d381c0f86fe10190dc3c3b5413\n",
|
|
||||||
"Successfully built iopath\n",
|
|
||||||
"Installing collected packages: iopath\n",
|
|
||||||
" Found existing installation: iopath 0.1.2\n",
|
|
||||||
" Uninstalling iopath-0.1.2:\n",
|
|
||||||
" Successfully uninstalled iopath-0.1.2\n",
|
|
||||||
"Successfully installed iopath-0.1.2\n"
|
|
||||||
],
|
|
||||||
"name": "stdout"
|
|
||||||
}
|
|
||||||
]
|
]
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
@@ -1984,13 +1893,26 @@
|
|||||||
}
|
}
|
||||||
]
|
]
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
"cell_type": "markdown",
|
||||||
|
"metadata": {
|
||||||
|
"id": "bxiRGTX3rR2-"
|
||||||
|
},
|
||||||
|
"source": [
|
||||||
|
"# Visualize\r\n",
|
||||||
|
"install opencv additinoally for visualize"
|
||||||
|
]
|
||||||
|
},
|
||||||
{
|
{
|
||||||
"cell_type": "code",
|
"cell_type": "code",
|
||||||
"metadata": {
|
"metadata": {
|
||||||
"id": "qRpGotxpzzZ-"
|
"id": "qRpGotxpzzZ-"
|
||||||
},
|
},
|
||||||
"source": [
|
"source": [
|
||||||
""
|
"!python demo/demo.py\\\r\n",
|
||||||
|
" --config-file projects/OneNet/configs/onenet.res50.dcn.yaml \\\r\n",
|
||||||
|
" --input path/to/images --output path/to/save_images --confidence-threshold 0.4 \\\r\n",
|
||||||
|
" --opts MODEL.WEIGHTS path/to/model.pth"
|
||||||
],
|
],
|
||||||
"execution_count": null,
|
"execution_count": null,
|
||||||
"outputs": []
|
"outputs": []
|
||||||
|
|||||||
Reference in New Issue
Block a user