有没有想过尝试用 Resnet、Densenet、Inception 等不同的块来搭建一个网络?
我假设你已经熟悉计算机视觉的基础知识。在深入研究之前,确保你知道什么是 Resnet,什么是 Densenet,什么是 Inception。
我们开始吧
- 安装
- 加载数据
- 创建并调试网络
- 列车
安装 Monk,一个低代码深度学习工具,也是计算机视觉的统一包装器。
$git clone [https://github.com/Tessellate-Imaging/monk_v1.git](https://github.com/Tessellate-Imaging/monk_v1.git)#Select the requirements file as per OS and CUDA version$cd monk_v1/installation/Linux && pip install -r requirements_cu9.txt
这里我们使用的是斯坦福狗分类数据集。
$! wget --load-cookies /tmp/cookies.txt "https://docs.google.com/uc?export=download&confirm=$(wget --save-cookies /tmp/cookies.txt --keep-session-cookies --no-check-certificate 'https://docs.google.com/uc?export=download&id=1b4tC_Pl1O80of7U-PJ7VExmszzSX3ZEM' -O- | sed -rn 's/.*confirm=([0-9A-Za-z_]+).*/**\1\n**/p')&id=1b4tC_Pl1O80of7U-PJ7VExmszzSX3ZEM" -O dogs-species-dataset.zip && rm -rf /tmp/cookies.txt
这里我们创建一个网络,并追加 resnet_v1_block 和 resnet_v2_block。
调试定制模型
$ gtf.debug_custom_model_design(network);
从下往上看
接下来,我们将追加 resnet _ v1 _ 瓶颈 _ 块和 resnet _ v2 _ 瓶颈 _ 块
调试定制模型
$ gtf.debug_custom_model_design(network);
从下往上看
这里我们将追加 densenet_block,inception_a_block 和 inception_c_block。
调试定制模型
$ gtf.debug_custom_model_design(network);
从下往上看
我们展平,添加一个完全连接的层,然后是一个完全连接的层,其中神经元(单元)的数量=自定义数据集中的类的数量。
调试定制模型
$ gtf.debug_custom_model_design(network);
从下往上看
设置时期、优化器、损耗和学习率调度程序。
你可以在 Github%20Create%20a%20custom%20network%20with%20blocks%20while%20debugging%20it-Copy1.ipynb) 上找到完整的 jupyter 笔记本。
我对计算机视觉和深度学习充满热情。我是 Monk 库的开源贡献者。
你也可以在以下网址看到我的其他作品:
medium.com](https://medium.com/@akulahemanth)