gaopengcuhk/Unofficial-Pix2Seq

How to inference in custom dataset?

YunLongPan opened this issue · 3 comments

I have a problem with the predict.py code of DETR

(detr) D:\Pix2Seq-main>python predict.py
dict_keys(['model', 'optimizer', 'lr_scheduler', 'epoch', 'args'])
Traceback (most recent call last):
File "predict.py", line 110, in
scores, boxes = predict(im, model, transform)
File "predict.py", line 85, in predict
outputs = model(data)
File "D:\Anaconda\envs\detr\lib\site-packages\torch\nn\modules\module.py", line 550, in call
result = self.forward(*input, **kwargs)
TypeError: forward() missing 1 required positional argument: 'sequence'

We will add a predict.py demo in the future. Stay finetuned.

We will add a predict.py demo in the future. Stay tuned.

We upload predict.py and also three demo object detection images. Thanks so much for your suggestion.