search

Went-Liang / UnSniffer

[CVPR 2023] Official Pytorch code for Unknown Sniffer for Object Detection: Don’t Turn a Blind Eye to Unknown Objects
https://arxiv.org/pdf/2303.13769v1.pdf
Apache License 2.0
67 stars 9 forks source link

Train on custom dataset? #7

Closed sf-pear closed 11 months ago

sf-pear commented 1 year ago

Hi there,

Can I get some guidelines on what I need to do to train UnSniffer on custom data? I'm trying out different OWOD approaches with deep sea images to see how well they perform.

Any help on what needs to be changed to train on custom data would be fantastic help. I'd be more than happy to return the favor by contributing to the documentation/code to allow other people to do the same :)

Thanks!

Went-Liang commented 1 year ago

You can try to add custom datasets in the form of other setup functions in setup_datasets.py. In addition, you need to add the basic information of the custom dataset in metadata.py.