Skip to content

Instantly share code, notes, and snippets.

@hugozanini

hugozanini/configuring-arch.ipynb

Created March 25, 2023 20:40
Show Gist options
  • Save hugozanini/b5f731d31c76564883ed8b1f704443ed to your computer and use it in GitHub Desktop.
Save hugozanini/b5f731d31c76564883ed8b1f704443ed to your computer and use it in GitHub Desktop.
Download ZIP
Display the source blob
Display the rendered blob
Raw
configuring-arch.ipynb
{
"nbformat": 4,
"nbformat_minor": 0,
"metadata": {
"colab": {
"provenance": []
},
"kernelspec": {
"name": "python3",
"display_name": "Python 3"
},
"language_info": {
"name": "python"
}
},
"cells": [
{
"cell_type": "code",
"execution_count": null,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "Q8-QJwPJu0RS",
"outputId": "f3377e2e-1e7c-41ee-9f8f-c8cde3ec25db"
},
"outputs": [
{
"output_type": "stream",
"name": "stdout",
"text": [
"Writing cfg/training/yolov7_stockout-tiny.yaml\n"
]
}
],
"source": [
"%%writefile cfg/training/yolov7_stockout-tiny.yaml\n",
"# parameters\n",
"nc: 1 # number of classes\n",
"depth_multiple: 1.0 # model depth multiple\n",
"width_multiple: 1.0 # layer channel multiple\n",
" \n",
"# anchors\n",
"anchors:\n",
" - [10,13, 16,30, 33,23] # P3/8\n",
" - [30,61, 62,45, 59,119] # P4/16\n",
" - [116,90, 156,198, 373,326] # P5/32\n",
" \n",
"# yolov7-tiny backbone\n",
"backbone:\n",
" # [from, number, module, args] c2, k=1, s=1, p=None, g=1, act=True\n",
" [[-1, 1, Conv, [32, 3, 2, None, 1, nn.LeakyReLU(0.1)]], # 0-P1/2 \n",
" \n",
" [-1, 1, Conv, [64, 3, 2, None, 1, nn.LeakyReLU(0.1)]], # 1-P2/4 \n",
" \n",
" [-1, 1, Conv, [32, 1, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-2, 1, Conv, [32, 1, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-1, 1, Conv, [32, 3, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-1, 1, Conv, [32, 3, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [[-1, -2, -3, -4], 1, Concat, [1]],\n",
" [-1, 1, Conv, [64, 1, 1, None, 1, nn.LeakyReLU(0.1)]], # 7\n",
" \n",
" [-1, 1, MP, []], # 8-P3/8\n",
" [-1, 1, Conv, [64, 1, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-2, 1, Conv, [64, 1, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-1, 1, Conv, [64, 3, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-1, 1, Conv, [64, 3, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [[-1, -2, -3, -4], 1, Concat, [1]],\n",
" [-1, 1, Conv, [128, 1, 1, None, 1, nn.LeakyReLU(0.1)]], # 14\n",
" \n",
" [-1, 1, MP, []], # 15-P4/16\n",
" [-1, 1, Conv, [128, 1, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-2, 1, Conv, [128, 1, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-1, 1, Conv, [128, 3, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-1, 1, Conv, [128, 3, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [[-1, -2, -3, -4], 1, Concat, [1]],\n",
" [-1, 1, Conv, [256, 1, 1, None, 1, nn.LeakyReLU(0.1)]], # 21\n",
" \n",
" [-1, 1, MP, []], # 22-P5/32\n",
" [-1, 1, Conv, [256, 1, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-2, 1, Conv, [256, 1, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-1, 1, Conv, [256, 3, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-1, 1, Conv, [256, 3, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [[-1, -2, -3, -4], 1, Concat, [1]],\n",
" [-1, 1, Conv, [512, 1, 1, None, 1, nn.LeakyReLU(0.1)]], # 28\n",
" ]\n",
" \n",
"# yolov7-tiny head\n",
"head:\n",
" [[-1, 1, Conv, [256, 1, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-2, 1, Conv, [256, 1, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-1, 1, SP, [5]],\n",
" [-2, 1, SP, [9]],\n",
" [-3, 1, SP, [13]],\n",
" [[-1, -2, -3, -4], 1, Concat, [1]],\n",
" [-1, 1, Conv, [256, 1, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [[-1, -7], 1, Concat, [1]],\n",
" [-1, 1, Conv, [256, 1, 1, None, 1, nn.LeakyReLU(0.1)]], # 37\n",
" \n",
" [-1, 1, Conv, [128, 1, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-1, 1, nn.Upsample, [None, 2, 'nearest']],\n",
" [21, 1, Conv, [128, 1, 1, None, 1, nn.LeakyReLU(0.1)]], # route backbone P4\n",
" [[-1, -2], 1, Concat, [1]],\n",
" \n",
" [-1, 1, Conv, [64, 1, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-2, 1, Conv, [64, 1, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-1, 1, Conv, [64, 3, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-1, 1, Conv, [64, 3, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [[-1, -2, -3, -4], 1, Concat, [1]],\n",
" [-1, 1, Conv, [128, 1, 1, None, 1, nn.LeakyReLU(0.1)]], # 47\n",
" \n",
" [-1, 1, Conv, [64, 1, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-1, 1, nn.Upsample, [None, 2, 'nearest']],\n",
" [14, 1, Conv, [64, 1, 1, None, 1, nn.LeakyReLU(0.1)]], # route backbone P3\n",
" [[-1, -2], 1, Concat, [1]],\n",
" \n",
" [-1, 1, Conv, [32, 1, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-2, 1, Conv, [32, 1, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-1, 1, Conv, [32, 3, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-1, 1, Conv, [32, 3, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [[-1, -2, -3, -4], 1, Concat, [1]],\n",
" [-1, 1, Conv, [64, 1, 1, None, 1, nn.LeakyReLU(0.1)]], # 57\n",
" \n",
" [-1, 1, Conv, [128, 3, 2, None, 1, nn.LeakyReLU(0.1)]],\n",
" [[-1, 47], 1, Concat, [1]],\n",
" \n",
" [-1, 1, Conv, [64, 1, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-2, 1, Conv, [64, 1, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-1, 1, Conv, [64, 3, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-1, 1, Conv, [64, 3, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [[-1, -2, -3, -4], 1, Concat, [1]],\n",
" [-1, 1, Conv, [128, 1, 1, None, 1, nn.LeakyReLU(0.1)]], # 65\n",
" \n",
" [-1, 1, Conv, [256, 3, 2, None, 1, nn.LeakyReLU(0.1)]],\n",
" [[-1, 37], 1, Concat, [1]],\n",
" \n",
" [-1, 1, Conv, [128, 1, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-2, 1, Conv, [128, 1, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-1, 1, Conv, [128, 3, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [-1, 1, Conv, [128, 3, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [[-1, -2, -3, -4], 1, Concat, [1]],\n",
" [-1, 1, Conv, [256, 1, 1, None, 1, nn.LeakyReLU(0.1)]], # 73\n",
" \n",
" [57, 1, Conv, [128, 3, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [65, 1, Conv, [256, 3, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" [73, 1, Conv, [512, 3, 1, None, 1, nn.LeakyReLU(0.1)]],\n",
" \n",
" [[74,75,76], 1, IDetect, [nc, anchors]], # Detect(P3, P4, P5)\n",
" ]"
]
}
]
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment