迁移学习分类图片

最新推荐文章于 2024-07-10 18:58:19 发布

Mike-H

最新推荐文章于 2024-07-10 18:58:19 发布

阅读量775

点赞数

分类专栏：深度学习

本文链接：https://blog.youkuaiyun.com/QLBFA/article/details/105880378

版权

基于tensorflow1.9+gpu可分类图片

import tensorflow as tf
import os
import numpy as np
import re
from PIL import Image
import matplotlib.pyplot as plt
lines=tf.gfile.GFile("/path/classify_mission/retrain/output_labels.txt").readlines()
uid_to_human={
   }

#一行一行读取数据,就字典的形式存储那些文件夹的名称（也就是下面分类的结果）
for uid,line in enumerate(lines):
    #去掉换行符
    line=line.strip('\n')
    uid_to_human[uid