amirbar / DETReg

Official implementation of the CVPR 2022 paper "DETReg: Unsupervised Pretraining with Region Priors for Object Detection".
https://amirbar.net/detreg
Apache License 2.0
334 stars 46 forks source link

Question about error in the GoogleColab demo #42

Closed s57224kn closed 2 years ago

s57224kn commented 2 years ago

Hi! I'm a beginner. I'm sorry for my poor English. Sorry, same question as #33.

Content

I did a demo on GoogleColab and received the following error in the 'Define model and load checkpoint' cell. How can I solve this problem? Or can I plot the results locally instead of GoogleColab?

It's resolved. Sorry for my lack of knowledge.

Error I've received

ModuleNotFoundError Traceback (most recent call last) in () 3 from PIL import Image 4 import requests ----> 5 from main import get_args_parser 6 from models import build_model 7 from argparse import Namespace

6 frames /content/gdrive/My Drive/DETReg/models/ops/functions/ms_deform_attn_func.py in () 16 from torch.autograd.function import once_differentiable 17 ---> 18 import MultiScaleDeformableAttention as MSDA 19 20

ModuleNotFoundError: No module named 'MultiScaleDeformableAttention'