tensorflow2
文章平均质量分 58
model2005
农技猿
展开
-
TensorFlow Lite Model Maker
TensorFlow Lite Model Maker 库,可以简化使用自定义数据集训练 TensorFlow Lite 模型的过程。该库使用迁移学习来减少所需的训练数据量并缩短训练时间。例,图像分类,四行代码# Load input data specific to an on-device ML app.data = ImageClassifierDataLoader.from_folder('flower_photos/')train_data, test_data = data.spl原创 2022-03-16 14:55:45 · 1957 阅读 · 0 评论 -
make_image_classifier 分类模型
make_image_classifier安装需安装tensorflow和tensorflow_hub libraries:$ pip install "tensorflow~=2.0"$ pip install "tensorflow-hub[make_image_classifier]~=0.6"测试:$ make_image_classifier --help最好安装GPU 版本TF2"tensorflow-gpu~=2.0".基本用法$ make_...翻译 2021-10-23 10:32:05 · 247 阅读 · 0 评论 -
图像分类模型训练image retrain
图片分类拷贝待训练的分类图片文件夹到tensorflow→tensorflow->examples->image_retraining目录下。如dm文件夹中的类别按各龄期为一个文件夹(注意查看下,文件夹中是否为各类别)训练python retrain.py --bottleneck_dir=model --how_many_training_steps 4000 --output_graph=model/retrained_graph.pb --output_la.原创 2021-10-22 11:18:45 · 478 阅读 · 0 评论 -
win10+GTX1060+tf1.15+object detection API踩坑记(5)
Non-OK-status: GpuLaunchKernel(FillPhiloxRandomKernelLaunch<Distribution>, num_blocks, block_size, 0, d.stream(), gen, data, size, dist) status: Internal: invalid device function查找资料,本故障的原因无外乎cuda和cudnn版本不配,或cuda版本过高。models-master版本过高?以2017年下载的版本原创 2021-09-22 09:52:57 · 406 阅读 · 0 评论 -
win10+GTX1060+tf1.15+object detection API踩坑记(4)
2021-09-08 06:53:46.096205: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1325] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 4863 MB memory) -> physical GPU (device: 0, name: GeForce GTX 1660 Ti, pci bus id: 0000:01:原创 2021-09-08 08:51:27 · 270 阅读 · 0 评论 -
tf2 进行目标检测训练
tf从0到tf1.x,现在已是tf2.x了,目标检测训练也得跟上新时代。object detection api配置基本差不多,数据集仍为VOC2007格式。根据要求把文件放到各自目录后。先把数据集转成训练需要的train.record和val.record格式。create_pascal_tf_record.pyarroplan_val.txt?? 什么原因??这么会有这个文件,No such file or directory.查看源文件,怎么有aeroplane的前缀?是b..原创 2021-08-19 16:00:03 · 230 阅读 · 0 评论