建议使用以下浏览器,以获得最佳体验。 IE 9.0+以上版本 Chrome 31+ 谷歌浏览器 Firefox 30+ 火狐浏览器
温馨提示

抱歉,您需设置社区昵称后才能参与社区互动!

前往修改
我再想想

求官方torch数据集转TensorFlow keras数据集脚本或数据集。

建行袁覃 2020/2/22 857

需求标注文件格式

xxx.jpg,0

xxx.jpg,1

xxx.jpg,2.......


回复 (1)

2020/2/23 15:22

keras 可以直接使用ImageDataGenerator读取数据,DATA_PATH=yourpath/images/,classes需要和那个txt从上到下写个list对应一下就可以了


from keras.preprocessing.image import ImageDataGenerator train_gen = ImageDataGenerator( rotation_range=20,validation_split=0.3) train_generator = train_gen.flow_from_directory(DATA_PATH, target_size=(224, 224),                                            classes=label_list, batch_size=batch_size,                                            subset='training')# 验证数据


validation_generator = train_gen.flow_from_directory(DATA_PATH, target_size=(224, 224),                                            classes=label_list, batch_size=batch_size,                                            subset='validation')

上划加载中
标签
您还可以添加5个标签
  • 没有搜索到和“关键字”相关的标签
  • 云产品
  • 解决方案
  • 技术领域
  • 通用技术
  • 平台功能
取消

建行袁覃

角色:成员

话题:4

发消息
发表于2020年02月22日 22:38:27 8571
直达本楼层的链接
楼主
倒序浏览 只看该作者
求官方torch数据集转TensorFlow keras数据集脚本或数据集。

需求标注文件格式

xxx.jpg,0

xxx.jpg,1

xxx.jpg,2.......


点赞 举报
分享

分享文章到朋友圈

分享文章到微博

混子选手

角色:成员

话题:1

发消息
发表于2020年02月23日 15:22:29
直达本楼层的链接
沙发
只看该作者

keras 可以直接使用ImageDataGenerator读取数据,DATA_PATH=yourpath/images/,classes需要和那个txt从上到下写个list对应一下就可以了


from keras.preprocessing.image import ImageDataGenerator train_gen = ImageDataGenerator( rotation_range=20,validation_split=0.3) train_generator = train_gen.flow_from_directory(DATA_PATH, target_size=(224, 224),                                            classes=label_list, batch_size=batch_size,                                            subset='training')# 验证数据


validation_generator = train_gen.flow_from_directory(DATA_PATH, target_size=(224, 224),                                            classes=label_list, batch_size=batch_size,                                            subset='validation')

点赞1 评论 引用 举报

游客

您需要登录后才可以回帖 登录 | 立即注册