开发者分享 |
您所在的位置:网站首页 › pytorch加载数据集 › 开发者分享 |
if os.path.exists(raw_data_path + '/'+ "labels_new.txt"): os.rename(raw_data_path + '/'+ "labels.txt", raw_data_path + '/'+ "labels_orig.txt") os.rename(raw_data_path + '/'+ "labels_new.txt", raw_data_path + '/'+ "labels.txt") @functools.lru_cache(1) def getSampleInfoList(raw_data_path): sample_list = [] with open(str(raw_data_path) + '/labels.txt', "r") as f: reader = csv.reader(f, delimiter = ' ') for i, row in enumerate(reader): imgname = row[0] label = int(row[1]) sample_list.append(DataInfoTuple(imgname, label)) sample_list.sort(reverse=False, key=myFunc) return sample_list del mydataset mydataset = MyDataset(isValSet_bool = None, raw_data_path = '../../raw_data/data_images', norm = False) len(mydataset) |
今日新闻 |
推荐新闻 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |