Zhang Zhonghao
Results
3
comments of
Zhang Zhonghao
![Uploading predict.jpg…]()
我检查了代码,问题再加载数据时,不是加载的所有数据集,是第一个循环的数据
数据集划分错误: 由于 read_split_data 函数中的 return 语句放置在第一个类别循环的末尾,导致函数只会处理第一个类别的数据并提前返回。确保数据集划分包含所有类别。 def read_split_data(root: str, val_rate: float = 0.2, plot_image: bool = False): # 保证随机结果可复现 random.seed(0) assert os.path.exists(root), f'dataset root {root} does not exist.' # 遍历文件夹,一个文件夹对应一个类别...