How can the network use Visual Genome classes to detect objects?

jwyang / faster-rcnn.pytorch

A faster pytorch implementation of faster r-cnn

MIT License

7.71k stars 2.33k forks source link

How can the network use Visual Genome classes to detect objects? #889

Open wihn2021 opened 2 years ago

wihn2021 commented 2 years ago

I have tried the default pascal_classes, and MS COCO classes by changing some settings and add the 80 classes of MS COCO to demo.py. How should I do to detect objects with Visual Genome categories?