09 2024 档案

深度学习（可视化特征图）
摘要：深度学习骨干网络一般会包含很多层，这里写了一个脚本，可以保存骨干网络的所有特征图。 代码主要用了get_graph_node_names和create_featrue_extractor这两个函数。 get_graph_node_names是得到所有特征节点名字。 create_featrue_ex 阅读全文
posted @ 2024-09-30 22:31 Dsp Tian 阅读(91) 评论(0) 推荐(0) 编辑
Linux查看命令动态输出
摘要：比如nvidia-smi命令会输出当前时刻显卡占用率，如果想动态监控，可以输入如下命令： watch -n 0.1 nvidia-smi 阅读全文
posted @ 2024-09-30 17:42 Dsp Tian 阅读(19) 评论(0) 推荐(0) 编辑
深度学习（FCN）
摘要：FCN是全卷积网络，用于做图像语义分割。通常将一般卷积网络最后的全连接层换成上采样或者反卷积网络，对图像的每个像素做分类，从而完成图像分割任务。 网络结构如下： 这里并没有完全按照原始网络结构实现，而是尝试upsample和convTranspose2d结合的方式，看看有什么效果。 下面代码是用VO 阅读全文
posted @ 2024-09-15 22:23 Dsp Tian 阅读(94) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示