
图像分类
文章平均质量分 53
搬运工学黛玛
这个作者很懒,什么都没留下…
展开
-
tensorflow2.0 densenet
import tensorflow as tffrom tensorflow.keras import Modeldef bottlenecklayer(filters,x): x=tf.keras.layers.BatchNormalization()(x) x=tf.nn.relu(x) x=tf.keras.layers.Conv2D(filters=filters, strides=1, kernel_size=1, padding="same")(x) x .原创 2021-06-29 21:07:35 · 484 阅读 · 0 评论 -
小卡片下的垃圾分类tensorflow2.0系列
垃圾分类纸片图是来自一些早教产品的,这样只通过截取纸片的垃圾比收集数据集更方便。我们选取5种垃圾展现情况,酒精、篮球、面包,帽子,还有空物下检测台的“十字架”,数据集下载链接:https://download.youkuaiyun.com/download/YANGJINCHAHAHA/19867998 将原始数据打乱并按照train-valid-test的文件结构分成三部分,split_dataset.py,代码假设你的源数据集在文件夹original_dataset,保存在dataset文件夹下。...原创 2021-06-28 11:52:46 · 366 阅读 · 0 评论 -
利用pytorch训练网络---垃圾分类,(resnet18)
数据集包含6种垃圾,分别为cardboard(纸箱),glass(玻璃)、metal(金属)、paper(纸)、plastic(塑料)、其他废品(trash),数据数量较小,仅供学习。数据集标准备工作,包括将数据集分为训练集和测试集,制作标签文件。代码utils.pyimport osimport shutilimport jsonpath="e://dataset//Garb...原创 2019-07-09 17:20:28 · 12529 阅读 · 73 评论