YifanXu74 / MQ-Det

Official PyTorch implementation of "Multi-modal Queried Object Detection in the Wild" (accepted by NeurIPS 2023)
Apache License 2.0
258 stars 12 forks source link

如何使用 Modulated pre-training后的模型来检测少量图片上的效果? #24

Open Mugealways opened 10 months ago

Mugealways commented 10 months ago

作者你好。我已经在coco数据集上抽取vision query并执行 Modulated pre-training了,假如我有少量的几张图片,想要知道训练后的模型对这几张图片的检测效果,就像您论文中figure4 中的图片一样,带有检测框和分数;请问我要执行什么命令来完成这一步呢?或者说我要怎么做可以让模型在几张少量的图片上做检测?期待您的回复,谢谢!

YifanXu74 commented 10 months ago

嗨,如果需要做可视化,可能需要自己写个demo了,近期比较忙,短期内可能没有时间更新demo。如果图片本身有标注,另外一种方式就是直接把这些图片的标注整成coco的标注格式,就可以直接参考readme里测lvis或者odinw的方式测试了

Mugealways commented 10 months ago

好的,谢谢!