loader迭代到最后,直接终止
收藏
我运行的是ppseg-release-2.5 ppLiteSeg源码,训练loader和config部分如下:
# ---------------- loader ----------------#
loader = ......
iter = start_iter
while iter < iters:
for data in loader:
iter+=1
print(iter,' ----- ',data[0].shape,data[1].shape)
'''
其余操作均注释掉
'''
# ---------------- config -----------------#
train_dataset:
type: Dataset
dataset_root: data/camvid
num_classes: 2
mode: train
train_path: data/camvid/train.txt
transforms:
- type: Normalize
val_dataset:
type: Dataset
dataset_root: data/camvid
num_classes: 2
mode: val
val_path: data/camvid/val.txt
transforms:
- type: Normalize
(1)win10
(2)我的图像尺寸全部一致(3,1024,256),标签(1024,256),分割目标一个类别;
(3)数据处理操作全部去掉,仅仅保留归一化;
(4)不管使用多少张图像,不管val部分是否关掉,都是迭代完一次 loader 后终止,不跳到下次loader迭代
比如:
iters=20000、1000张图、bs=2:输出 500,' ----- ',[2,3,256,1024],[2,256,1024],然后终止
iters=20000、10张图、bs=2:输出 5,' ----- ',[2,3,256,1024],[2,256,1024],然后终止
请问有人碰到过这个问题吗?
0
收藏
请登录后评论