search

ruhyadi / YOLO3D

YOLO 3D Object Detection for Autonomous Driving Vehicle
https://ruhyadi.github.io/project/computer-vision/yolo3d
247 stars 43 forks source link

Adjustments to make it running on Colab #34

Open timspilak opened 2 months ago

timspilak commented 2 months ago

It wasn't running in colab, so I had to make the following adjustments:

inference.py line 40 &41: 

    'resnet': resnet18(weights='IMAGENET1K_V1'),
    'resnet18': resnet18(weights='IMAGENET1K_V1'),

inference.py line 74:

checkpoint = torch.load(reg_weights, weights_only=True)`

library/Math.py", line 163:

A = np.zeros([4,3], dtype=np.float64)