search

shaoshengsong / MobileNetV3-SSD

MobileNetV3-SSD for object detection and implementation in PyTorch
269 stars 88 forks source link

请问这个怎么改？ #6

Open duanyuqi987 opened 5 years ago

duanyuqi987 commented 5 years ago

Traceback (most recent call last): File "run_ssd_live_demo.py", line 73, in boxes, labels, probs = predictor.predict(image, 10, 0.4) File "/home/yajun/github/MobileNetV3-SSD/vision/ssd/predictor.py", line 37, in predict scores, boxes = self.net.forward(images) File "/home/yajun/github/MobileNetV3-SSD/vision/ssd/ssd.py", line 100, in forward locations, self.priors, self.config.center_variance, self.config.size_variance File "/home/yajun/github/MobileNetV3-SSD/vision/utils/box_utils.py", line 104, in convert_locations_to_boxes locations[..., :2] center_variance priors[..., 2:] + priors[..., :2], RuntimeError: expected backend CPU and dtype Float but got backend CUDA and dtype Float

lingyi-learning commented 5 years ago

同问

shaoshengsong commented 4 years ago

It is recommended that git clone this version https://github.com/shaoshengsong/MobileNetV3-SSD-Compact-Version