Hello,
On my jetson TX1 I have been trying to run a code to do some stereo object detection. The code works fine on my pc, but when I set up all my dependency and I dowload pythorch for aarm linux forn this github : GitHub - KumaTea/pytorch-aarch64: PyTorch wheels (whl) & conda for aarch64 / ARMv8 / ARM64 on my jetson TX1. I keep getting this error :
nvidia@ubuntu:~/Documents/depth$ /usr/local/bin/python3.7 /home/nvidia/Documents/depth/stereoVision.py
Downloading: “https://github.com/ultralytics/yolov5/archive/master.zip" to /home/nvidia/.cache/torch/hub/master.zip
/usr/local/lib/python3.7/site-packages/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension:
warn(f"Failed to load image Python extension: {e}”)
Traceback (most recent call last):
File “/home/nvidia/Documents/depth/stereoVision.py”, line 17, in
model = torch.hub.load(‘ultralytics/yolov5’, ‘custom’, path=‘/home/nvidia/Documents/projet/yolov5/runs/train/exp3/weights/last.pt’, force_reload=True)
File “/usr/local/lib/python3.7/site-packages/torch/hub.py”, line 399, in load
model = _load_local(repo_or_dir, model, *args, **kwargs)
File “/usr/local/lib/python3.7/site-packages/torch/hub.py”, line 428, in _load_local
model = entry(*args, **kwargs)
File “/home/nvidia/.cache/torch/hub/ultralytics_yolov5_master/hubconf.py”, line 70, in custom
return _create(path, autoshape=autoshape, verbose=verbose, device=device)
File “/home/nvidia/.cache/torch/hub/ultralytics_yolov5_master/hubconf.py”, line 31, in _create
from models.common import AutoShape, DetectMultiBackend
File “/home/nvidia/.cache/torch/hub/ultralytics_yolov5_master/models/common.py”, line 24, in
from utils.datasets import exif_transpose, letterbox
File “/home/nvidia/.cache/torch/hub/ultralytics_yolov5_master/utils/datasets.py”, line 30, in
from utils.augmentations import Albumentations, augment_hsv, copy_paste, letterbox, mixup, random_perspective
File “/home/nvidia/.cache/torch/hub/ultralytics_yolov5_master/utils/augmentations.py”, line 12, in
from utils.general import LOGGER, check_version, colorstr, resample_segments, segment2box
File “/home/nvidia/.cache/torch/hub/ultralytics_yolov5_master/utils/general.py”, line 30, in
import torchvision
File “/usr/local/lib/python3.7/site-packages/torchvision/init.py”, line 7, in
from torchvision import models
File “/usr/local/lib/python3.7/site-packages/torchvision/models/init.py”, line 18, in
from . import quantization
File “/usr/local/lib/python3.7/site-packages/torchvision/models/quantization/init.py”, line 1, in
from .mobilenet import *
File “/usr/local/lib/python3.7/site-packages/torchvision/models/quantization/mobilenet.py”, line 1, in
from .mobilenetv2 import QuantizableMobileNetV2, mobilenet_v2, all as mv2_all
File “/usr/local/lib/python3.7/site-packages/torchvision/models/quantization/mobilenetv2.py”, line 5, in
from torch.ao.quantization import QuantStub, DeQuantStub
ImportError: cannot import name ‘QuantStub’ from ‘torch.ao.quantization’ (/usr/local/lib/python3.7/site-packages/torch/ao/quantization/init.py)
Does any one know how to fix this bug.
Thanks