将训练集构建成ImageNet模型

以下程序实现将训练集构建为ImageNet模型，训练集图片为56个民族

import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.PrintWriter;

public class createTxt {
    public static void createClassInd(){//56个民族编号及名称
        FileOutputStream fout;
        try {
            fout = new FileOutputStream("E:\\resourse\\InnovationFund\\file\\classInd.txt");//保存在该文件中
            PrintWriter pwr = new PrintWriter(fout);
            for (int i = 1; i < 57; i++) {
                pwr.print(i);
                pwr.print(" ");
                switch(i){
                case 1:
                    pwr.print("han");
                    break;
                case 2:
                    pwr.print("hui");
                    break;
                case 3:
                    pwr.print("zhuang");
                    break;
                case 4:
                    pwr.print("zang");
                    break;
                case 5:
                    pwr.print("yao");
                    break;
                case 6:
                    pwr.print("yugu");
                    break;
                case 7:
                    pwr.print("yi");
                    break;
                case 8:
                    pwr.print("xibo");
                    break;
                case 9:
                    pwr.print("weiwuer");
                    break;
                case 10:
                    pwr.print("li");
                    break;
                case 11:
                    pwr.print("wa");
                    break;
                case 12:
                    pwr.print("tujia");
                    break;
                case 13:
                    pwr.print("tu");
                    break;
                case 14:
                    pwr.print("tataer");
                    break;
                case 15:
                    pwr.print("tajike");
                    break;
                case 16:
                    pwr.print("shui");
                    break;
                case 17:
                    pwr.print("sala");
                    break;
                case 18:
                    pwr.print("qiang");
                    break;
                case 19:
                    pwr.print("pumi");
                    break;
                case 20:
                    pwr.print("nu");
                    break;
                case 21:
                    pwr.print("naxi");
                    break;
                case 22:
                    pwr.print("miao");
                    break;
                case 23:
                    pwr.print("menggu");
                    break;
                case 24:
                    pwr.print("menba");
                    break;
                case 25:
                    pwr.print("maonan");
                    break;
                case 26:
                    pwr.print("man");
                    break;
                case 27:
                    pwr.print("luoba");
                    break;
                case 28:
                    pwr.print("keerkezi");
                    break;
                case 29:
                    pwr.print("wuzibieke");
                    break;
                case 30:
                    pwr.print("she");
                    break;
                case 31:
                    pwr.print("mulao");
                    break;
                case 32:
                    pwr.print("susu");
                    break;
                case 33:
                    pwr.print("lagu");
                    break;
                case 34:
                    pwr.print("jinuo");
                    break;
                case 35:
                    pwr.print("jing");
                    break;
                case 36:
                    pwr.print("hezhe");
                    break;
                case 37:
                    pwr.print("hasake");
                    break;
                case 38:
                    pwr.print("hani");
                    break;
                case 39:
                    pwr.print("qilao");
                    break;
                case 40:
                    pwr.print("gaoshan");
                    break;
                case 41:
                    pwr.print("ewenke");
                    break;
                case 42:
                    pwr.print("eluosi");
                    break;
                case 43:
                    pwr.print("elunchun");
                    break;
                case 44:
                    pwr.print("dulong");
                    break;
                case 45:
                    pwr.print("dongxiang");
                    break;
                case 46:
                    pwr.print("tong");
                    break;
                case 47:
                    pwr.print("deang");
                    break;
                case 48:
                    pwr.print("dai");
                    break;
                case 49:
                    pwr.print("dawoer");
                    break;
                case 50:
                    pwr.print("chaoxian");
                    break;
                case 51:
                    pwr.print("buyi");
                    break;
                case 52:
                    pwr.print("bulang");
                    break;
                case 53:
                    pwr.print("baoan");
                    break;
                case 54:
                    pwr.print("bai");
                    break;
                case 55:
                    pwr.print("achang");
                    break;
                case 56:
                    pwr.print("jingpo");
                    break;
                }

                pwr.print("\r\n");//换行
            }
            pwr.close();
        }catch(Exception e){System.out.println("ClassInd。txt失败");}
    }
    public static boolean noDb(String str){//不包含返回true，判断文件是否为.db文件
        boolean in = true;
        if(str.indexOf(".db")!=-1)//包含
            in=false;
        return in;
    }

    private static void getFile(String path, PrintWriter pwr,PrintWriter tpwr,int n) throws FileNotFoundException{
        //测试集：训练集：验证集=6:2:2，访问目录下文件的文件名
        File file = new File(path);   //图片集文件夹
        // 获得该文件夹内的所有文件
        File[] array = file.listFiles();   //目录下所有文件名
        for(int i=0;i<array.length;i++)
        {
            if(array[i].isFile())//如果是文件
            {
                int fn=array.length*3/5;//6成
                int tfn = array.length*4/5;//2成
                // 只输出文件名字 

                 if(i < fn && noDb(array[i].getName())){//测试集
                     //System.out.println( array[i].getName()+" "+fn);
                     pwr.print(array[i].getName());
                     pwr.print(" ");
                     pwr.print(n);   //编号
                     pwr.print("\r\n");
                     //System.out.println( array[i].getPath());
                     //移动图片到指定目录
                     File afile = new File(array[i].getPath());
                   afile.renameTo(new File("E:\\resourse\\InnovationFund\\file\\test\\" + afile.getName()));
                 }else if(i < tfn && noDb(array[i].getName())){//训练集
                     tpwr.print(array[i].getName());
                     tpwr.print("  ");
                     tpwr.print(n);
                     tpwr.print("\r\n");
                     File afile = new File(array[i].getPath());
                    afile.renameTo(new File("E:\\resourse\\InnovationFund\\file\\trian\\" + afile.getName()));
                 }

            }
            else if(array[i].isDirectory())//如果是文件夹
            {
                System.out.println( array[i].getName());
                n++;
                getFile(array[i].getPath(), pwr, tpwr, n);  //递归

            }
        }

    }   

    public static void main(String args[]) throws FileNotFoundException {
        createClassInd();

        String path = "E:\\一些资料\\创新基金\\民族服饰 cai";
        FileOutputStream fout;
        fout = new FileOutputStream("E:\\resourse\\InnovationFund\\file\\test_label.txt");
        PrintWriter pwr = new PrintWriter(fout);

        FileOutputStream tfout;
        tfout = new FileOutputStream("E:\\resourse\\InnovationFund\\file\\train_label.txt");
        PrintWriter tpwr = new PrintWriter(tfout);

        int n=0;
        getFile(path, pwr,tpwr, n);
    }
}