Open eloisacorti opened 11 months ago
👋 Hello @eloisacorti, thank you for leaving an issue on Roboflow Notebooks.
If you are filing a bug report, please be as detailed as possible. This will help us more easily diagnose and resolve the problem you are facing. To learn more about contributing, check out our Contributing Guidelines.
If you require support with custom code that is not part of Roboflow Notebooks, please reach out on the Roboflow Forum or on the GitHub Discussions page associated with this repository.
Do you have more questions about Roboflow that we haven't responded to yet? Feel free to ask them on the Roboflow Discuss forum. Our developer advocates and community team actively respond to questions there.
To ask questions about Notebooks, head over to the GitHub Discussions section of this repository.
Hi @eloisacorti 👋🏻! Does that happen with football-players-detection-3zvbc
dataset? Or your custom one?
Hi @SkalskiP It happen with custom dataset.
@eloisacorti, is that dataset coming from Roboflow? If so, could you share a link?
No, I have it my drive.
The directory structure is:
maize10----|valid #images + .json |train #images + .json |__test #images + .json
https://drive.google.com/drive/folders/1d0LUp_SPpcbPdRVYvO3lHZjTTlJJaPWj?usp=sharing
Could you share you colab where you try and fail to load that dataset?
Yes. I only changed the location of the dataset https://colab.research.google.com/drive/1qIxmCAueQxkYd2b5Y6DAA-pWWVGuhskF?usp=sharing
Okey. I'll take a look at it as soon as possible. But I'm afraid I won't be able to do it this week.
Thanks!! @SkalskiP
@eloisacorti have you find a solution ? I got the same problem, for some datasets it throws the exact error as you encountered but for other datasets it works fine !
@eloisacorti @eloisacorti
change this line in your colab in the section causing the error:
results = image_processor.post_process_object_detection(outputs, target_sizes=orig_target_sizes)
to:
results = image_processor.post_process_object_detection(outputs, target_sizes=orig_target_sizes, threshold=0)
I think this is because the default threshold is too high that causes the results to be empty in most values.
@poyingHAHA so that happens because during evaluation, there are no detections produced by the model?
Yes, training step is fine, but the you get nothing during evaluation because of the default threshold and model not training well.
how do i ensure that the model is training well? like i am using this notebook as a reference to train for the object detection on soccernet dataset, and i am facing the same issue.
Hi folks,
Thanks for reporting, I was fine-tuning DETR and the new DETA model on a custom dataset and faced the same issue. Glad I found this 😅 will update my demo notebooks.
Regarding making sure the model is training well, fine-tuning is working well for me on the balloon dataset: https://github.com/NielsRogge/Transformers-Tutorials/blob/master/DETA/Fine_tuning_DETA_on_a_custom_dataset_(balloon).ipynb.
Also pushed an update for my DETR notebook.
Search before asking
Notebook name
train-huggingface-detr-on-custom-dataset.ipynb
Bug
When training on custom dataset, it seems like I'm not getting any predictions. During trainning I get loss resultos, so it seems like the trainning is doing fine, but then I can't get inferences. If I try to run the CocoEvaluator I get this bug:
`KeyError Traceback (most recent call last)