search

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
336 stars 46 forks source link

How to run DETReg on video custom dataset ? #48

Closed lemyanh2612 closed 1 year ago

lemyanh2612 commented 2 years ago

Could you provide me some feedback about this possibility and some guidance on where is the video custom data loading actually happening so that I could check?

Thank you for your time

amirbar commented 2 years ago

Thanks for your interest in our work. In this work, we mainly focused on static images and not videos, although it may be possible to extend this for videos. Are you interested in training DETReg on videos or just running inference on video?

lemyanh2612 commented 2 years ago

Thank you for your response. I'm working with the sonar dataset, including the grayscale videos for fish detection Source.

I have tried some models but the results are not very good. This link is an example of a sonar dataset. Could you give me some advice?

Thank you for your support.

amirbar commented 1 year ago

The most naive idea would be to perform video colorization then run the system as is.