DETR error with using own datasets (TypeError: Detections.__getitem__ not supported for index of type <class 'numpy.ndarray'>.)

[Yamiloveme]

Search before asking

• I have searched the Supervision issues and found no similar feature requests.

Question

I have used my own dataset and got annotations using roboflow.
I got this error can you please resolve this
--> 243 raise TypeError(
244 f"Detections.getitem not supported for index of type {type(index)}."
245 )

TypeError: Detections.getitem not supported for index of type <class 'numpy.ndarray'>.

I just change the dataset to durian detection dataset which is found from roboflow universe.
https://universe.roboflow.com/princess-chulabhorn-science-high-school-nakhon-si-thammarat/duriandetection

All the code remain the same except the dataset code.
https://colab.research.google.com/drive/1Ck8gCfZWbvNwk6C-VJTiyqaYkEueasxg?usp=sharing

Additional

No response

[github-actions[bot]]

Hello there, thank you for opening an Issue ! 🙏🏻 The team was notified and they will get back to you asap.

[LinasKo]

❗ ❗ ❗

For those stumbling upon this, check out https://colab.research.google.com/drive/1bg95DuiQA8EZt4QRymk3PxtseSU5Jkwa#scrollTo=Ts6bI87iTZzY

We shall be updating the notebook with this one in supervision 0.22.0

[onuralpszr]

@Yamiloveme for first hello and could you change supervision version to latest

!pip install -i https://test.pypi.org/simple/ supervision==0.3.0. you were using pretty old one.

As much as I wanted to help you, it would be better you switch to latest version so we can test and if possible share the collab you "used" with errors and etc. So we can also see easier and help us to understand better.

[SkalskiP]

Hi @Yamiloveme 👋🏻! I think @onuralpszr is right! Try to update supervision==0.13.0. And let us know if it worked.

In the meantime, I will also convert that issue into a discussion and move it into the Q/A section.

[dokooh]

Hi,

I have been running to this issue with either versions of supervision suggested (i.e. supervision==0.13.0 and supervision==0.3.0).

I realized that the error is thrown when there is no detection in the result set. So when the result is empty this error is thrown, perhaps something to consider for further testing and future releases.

[r0ss-mYstery]

Hi,
I encountered the same issue, did you fix that?

[SkalskiP]

What version of supervision are you using?

[r0ss-mYstery]

I am using the supervision==0.3.0, just following the custom dataset DETR tutorial from roboflow

[TianzhangZhao]

Hi,

I also encountered the same issue, and I'm using supervision==0.13.0

[LinasKo]

Hi @levi-hoang-CLV, @TianzhangZhao,

Does this still happen in more recent versions of supervision? 0.13.0 is very old - we recommend either supervision or supervision==0.21.0.

I recognize that the tutorial is a bit outdated in this regard.

[levi-hoang-CLV]

I tried two version of supervision 0.3.0 and 0.21.0. But it still not work for me!

[LinasKo]

I'll look into it.

[LinasKo]

Hi @levi-hoang-CLV, @TianzhangZhao,

Here's a temporary solution that fixes the reported error, but there's likely more.
https://drive.google.com/file/d/1JnUcL2x4MBIpGJRGvhCMAkDSyKEy28m_/view?usp=sharing

[LinasKo]

Here's the modernized train-huggingface-detr-on-custom-dataset.ipynb notebook:
https://colab.research.google.com/drive/1bg95DuiQA8EZt4QRymk3PxtseSU5Jkwa?usp=sharing

We'll add it to supervision in the next release.

[LinasKo]

For those stumbling upon this, check out https://colab.research.google.com/drive/1bg95DuiQA8EZt4QRymk3PxtseSU5Jkwa#scrollTo=Ts6bI87iTZzY

We shall be updating the notebook with this one in supervision 0.22.0