zl程序教程

您现在的位置是:首页 >  硬件

当前栏目

机器学习笔记 - 使用Keras、TensorFlow框架进行自定义数据集目标检测训练

机器训练笔记学习框架 进行 检测 目标
2023-09-14 09:01:35 时间

        学习如何训练自定义深度学习模型,以通过Keras和TensorFlow的来执行对象检测。

一、安装环境

        1、首先安装tensorflow 2.0,参考:机器学习笔记 - win10安装tensorflow-gpu.2.2 + Cuda10+cudnn7.6.5_bashendixie5的博客-CSDN博客https://blog.csdn.net/bashendixie5/article/details/110260615

        2、提前下载vgg16_weights_tf_dim_ordering_tf_kernels_notop.h5文件,放到C:\Users\【user】\.keras\models文件夹下

二、准备数据集

        1、首先安装打标签的工具,此次使用的是labelImg,git官网