RemoteSensing遥感影像分割,多波段train_demo.py报错
Created by: KuntaHu
您好,按照教程,将多波段转为npy格式保存好。同时配置好了数据文件格式。 因为输入有6个波段所以 训练时候 设置为--channel 6,然后运行train_demo.py 有如下报错:
2020-06-19 23:58:38 [INFO] 40 samples in file data/dataset/train.txt 2020-06-19 23:58:38 [INFO] 30 samples in file data/dataset/val.txt W0619 23:58:39.167647 1220 device_context.cc:237] Please NOTE: device: 0, CUDA Capability: 70, Driver API Version: 10.1, Runtime API Version: 9.0 W0619 23:58:39.171162 1220 device_context.cc:245] device: 0, cuDNN Version: 7.3. 2020-06-19 23:58:40,660-INFO: Instantiated empty configuration. HDFS initialization failed, please check if .hdfscli,cfg exists. Exception in thread Thread-6: Traceback (most recent call last): File "/opt/conda/envs/python35-paddle120-env/lib/python3.7/threading.py", line 926, in _bootstrap_inner self.run() File "/opt/conda/envs/python35-paddle120-env/lib/python3.7/threading.py", line 870, in run self._target(*self._args, **self._kwargs) File "/home/aistudio/contrib/RemoteSensing/readers/base.py", line 85, in handle_worker r = mapper(sample[0], sample[1], sample[2]) File "/home/aistudio/contrib/RemoteSensing/transforms/transforms.py", line 68, in call outputs = op(im, im_info, label) File "/home/aistudio/contrib/RemoteSensing/transforms/transforms.py", line 488, in call im = normalize(im, self.min_val, self.max_val, mean, std) File "/home/aistudio/contrib/RemoteSensing/transforms/ops.py", line 25, in normalize im = (im.astype(np.float32, copy=False) - min_value) / range_value ValueError: operands could not be broadcast together with shapes (256,256,6) (3,)
后来尝试选取3个波段,保存为npy格式,将--channel设置为3,重新读取数据,运行正确。说明我的数据文件格式是正确的。难道train_demo.py只能设定3波段输入?请问如何输入多波段呢?
谢谢!