tensorflow之使用shell脚本定义自己的图片标签

最新推荐文章于 2022-06-05 18:03:14 发布

csuzhaoqinghui

最新推荐文章于 2022-06-05 18:03:14 发布

阅读量2.7k

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.csdn.net/csuzhaoqinghui/article/details/51386542

本文介绍了一种使用Shell脚本自动为深度学习项目中的图片数据集打上分类标签的方法，该方法通过递归遍历文件夹并提取路径作为标签。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一直比较欣赏外国人的不管数据工作如何简单都会使用代码来操作数据。

学习谷歌的深度学习tensorflow一直认为给自己的数据库图片打上分类标签是一种费力事。

所以就尝试着写一些shell脚本来进行数据的脚本操作。以下代码是我为深度学习自动标签分类的代码。

############################################
#! /bin/bash -
#Author:zhaoqinghui
#Data  :2016.05.11 
#============ get the file name =========== 
INIT_PATH="./image";
#==========================================

function ergodic(){
  for file in `ls $1`
  do
    if [ -d $1"/"$file ]
    then
      ergodic $1"/"$file
    else
      local path=$1"/"$file 
      local name=$file      
      local size=`du --max-depth=1 $path|awk '{print $1}'` 
      #echo $name  $size $path 
      label=`echo ${path%/*}`
      #label=$`echo ${label%/*}`
      #label=$`echo ${label#*/}`
      label=`echo ${label#*/}`
      label=`echo ${label#*/}`
      echo  $path $label >> mytrain.txt
      echo 'save succuss'  $path
    fi
  done
}
IFS=$'\n' #防止有空格时出错
ergodic $INIT_PATH