1 Star 1 Fork 0

luotianhang/dog10分类-pytorch

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
preparedata.py 456 Bytes
一键复制 编辑 原始数据 按行查看 历史
罗天杭 提交于 2021-08-02 01:50 . 20210802
import os
dict = {"beagle": 0, "dachshund": 1, "dalmatian": 2, "jindo": 3, "maltese": 4, "pomeranian": 5, "retriever": 6,
"ShihTzu": 7, "toypoodle": 8, "Yorkshireterrier": 9}
path = "G:/datasets/分类/dog_class_num=10/val"
f = open("dog/test.txt", 'w', encoding='utf-8')
for file in os.listdir(path):
for image in os.listdir(os.path.join(path, file)):
f.write(str(dict[file])+' '+os.path.join(path,file,image))
f.write('\n')
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/luotianhang/luo-tianhang-code.git
git@gitee.com:luotianhang/luo-tianhang-code.git
luotianhang
luo-tianhang-code
dog10分类-pytorch
main

搜索帮助

0d507c66 1850385 C8b1a773 1850385