(base) wheeltec-client@ubuntu:~/Desktop/ggcnn$ source activate py36
(py36) wheeltec-client@ubuntu:~/Desktop/ggcnn$ python3 train_ggcnn.py
INFO:root:Loading Dataset...
>> train dataset: 86
>> train_val dataset: 8
>> test dataset: 9
INFO:root:Loading Network...
INFO:root:optimizer Done
----------------------------------------------------------------
Layer (type) Output Shape Param #
================================================================
Conv2d-1 [-1, 16, 360, 360] 1,952
ReLU-2 [-1, 16, 360, 360] 0
Conv2d-3 [-1, 16, 360, 360] 6,416
ReLU-4 [-1, 16, 360, 360] 0
MaxPool2d-5 [-1, 16, 180, 180] 0
Conv2d-6 [-1, 16, 180, 180] 6,416
ReLU-7 [-1, 16, 180, 180] 0
Conv2d-8 [-1, 16, 180, 180] 6,416
ReLU-9 [-1, 16, 180, 180] 0
MaxPool2d-10 [-1, 16, 90, 90] 0
Conv2d-11 [-1, 32, 90, 90] 12,832
ReLU-12 [-1, 32, 90, 90] 0
Conv2d-13 [-1, 32, 90, 90] 25,632
ReLU-14 [-1, 32, 90, 90] 0
UpsamplingBilinear2d-15 [-1, 32, 180, 180] 0
Conv2d-16 [-1, 16, 180, 180] 4,624
ReLU-17 [-1, 16, 180, 180] 0
UpsamplingBilinear2d-18 [-1, 16, 360, 360] 0
Conv2d-19 [-1, 16, 360, 360] 2,320
ReLU-20 [-1, 16, 360, 360] 0
Conv2d-21 [-1, 1, 360, 360] 17
Conv2d-22 [-1, 1, 360, 360] 17
Conv2d-23 [-1, 1, 360, 360] 17
Conv2d-24 [-1, 1, 360, 360] 17
================================================================
Total params: 66,676
Trainable params: 66,676
Non-trainable params: 0
----------------------------------------------------------------
Input size (MB): 0.49
Forward/backward pass size (MB): 159.19
Params size (MB): 0.25
Estimated Total Size (MB): 159.94
----------------------------------------------------------------
INFO:root:Beginning Epoch 00, lr=0.001
[ WARN:0@0.765] global loadsave.cpp:244 findDecoder imread_('test/pcd0150d.tiff'): can't open/read file: check file path/integrity
[ WARN:0@0.772] global loadsave.cpp:244 findDecoder imread_('test/pcd0185d.tiff'): can't open/read file: check file path/integrity
[ WARN:0@0.782] global loadsave.cpp:244 findDecoder imread_('test/pcd0113d.tiff'): can't open/read file: check file path/integrity
Traceback (most recent call last):
File "train_ggcnn.py", line 323, in <module>
run()
File "train_ggcnn.py", line 273, in run
train_results = train(epoch, net, device, train_data, optimizer)
File "train_ggcnn.py", line 142, in train
for x, y in train_data:
File "/home/wheeltec-client/anaconda3/envs/py36/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 521, in __next__
data = self._next_data()
File "/home/wheeltec-client/anaconda3/envs/py36/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 1203, in _next_data
return self._process_data(data)
File "/home/wheeltec-client/anaconda3/envs/py36/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 1229, in _process_data
data.reraise()
File "/home/wheeltec-client/anaconda3/envs/py36/lib/python3.6/site-packages/torch/_utils.py", line 434, in reraise
raise exception
AttributeError: Caught AttributeError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/home/wheeltec-client/anaconda3/envs/py36/lib/python3.6/site-packages/torch/utils/data/_utils/worker.py", line 287, in _worker_loop
data = fetcher.fetch(index)
File "/home/wheeltec-client/anaconda3/envs/py36/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 49, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/wheeltec-client/anaconda3/envs/py36/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 49, in <listcomp>
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/wheeltec-client/Desktop/ggcnn/utils/data/grasp_data.py", line 68, in __getitem__
image.rescale(scale)
File "/home/wheeltec-client/Desktop/ggcnn/utils/data/structure/img.py", line 165, in rescale
self.img = mmcv.imrescale(self.img, scale, interpolation=interpolation)
File "/home/wheeltec-client/Desktop/ggcnn/utils/dataset_processing/mmcv.py", line 130, in imrescale
h, w = img.shape[:2]
AttributeError: 'NoneType' object has no attribute 'shape'
(py36) wheeltec-client@ubuntu:~/Desktop/ggcnn$
运行train_ggcnn.py请问这是肿魔回事
评论(1)
您还未登录,请登录后发表或查看评论
第三方账号登入
QQ 微博 微信