训练问题

[tuqiangwang]

你好，数据集用的是HED-BSDS_PASCAL
执行python train.py。报以下错误：（TensorFlow1.6 python3，另外博主的环境是什么）
Traceback (most recent call last):
File "train.py", line 103, in
dataset = DataSet()
File "train.py", line 24, in init
self.read_data()
File "train.py", line 40, in read_data
self.imgs = np.zeros((len(img_names), self.cfg['height'], self.cfg['width'], self.cfg['channel']), np.float32)
MemoryError

[whyguu]

tf1.8 python3.5