kuangliu / pytorch-retinanet

RetinaNet in PyTorch
992 stars 250 forks source link

Anyone can help me to solve this problem? #64

Open BCWang93 opened 5 years ago

BCWang93 commented 5 years ago

I have this error when run train.py: Traceback (most recent call last): File "H:/PyCharm project/retinanet/train.py", line 101, in train(epoch) File "H:/PyCharm project/retinanet/train.py", line 55, in train for batch_idx, (inputs, loc_targets, cls_targets) in enumerate(trainloader): File "D:\Python\lib\site-packages\torch\utils\data\dataloader.py", line 582, in next return self._process_next_batch(batch) File "D:\Python\lib\site-packages\torch\utils\data\dataloader.py", line 608, in _process_next_batch raise batch.exc_type(batch.exc_msg) RuntimeError: Traceback (most recent call last): File "D:\Python\lib\site-packages\torch\utils\data_utils\worker.py", line 99, in _worker_loop samples = collate_fn([dataset[i] for i in batch_indices]) File "H:\PyCharm project\retinanet\utils\util.py", line 28, in collate_fn loc_target, cls_target = encode_target(boxes[i], labels[i], input_size=(w,h)) File "H:\PyCharm project\retinanet\utils\util.py", line 39, in encode_target anchor_bboxes=enumerate_all_anchors(input_size) File "H:\PyCharm project\retinanet\utils\util.py", line 64, in enumerate_all_anchors feature_map_size=[(input_size/pow(2.,i+3)).ceil() for i in range(num_feature)] # p3-p7feature_map size File "H:\PyCharm project\retinanet\utils\util.py", line 64, in feature_map_size=[(input_size/pow(2.,i+3)).ceil() for i in range(num_feature)] # p3-p7feature_map size RuntimeError: ceil_vml_cpu not implemented for 'Long'

Process finished with exit code 1 Anyone can help me?Thanks