vgg16的tf2实现

it2025-10-12  9

@[TOC]前言 学习经典卷积神经网络,在这里记录学习历程(经验与坑) 主要参考文章:原博客 原博客是基于tf1和keras实现的,这里修改代码使得能在tf2.2下运行。同时对代码进行详细注释,方便日后回顾。

源代码和数据集 源代码链接: https://pan.baidu.com/s/1rnv_ieni4srvGgS-TreZAQ 提取码: bnsq 数据集链接: https://pan.baidu.com/s/1ErZtyhlGRvAfevRqAQT2Ug 提取码: xrva

将数据集下载后解压到data/image/train,运行train.py即可开始训练,运行predict.py可以预测。

最新回复(0)