Running on Custom Dataset

jwyang / graph-rcnn.pytorch

[ECCV 2018] Official code for "Graph R-CNN for Scene Graph Generation"

732 stars 157 forks source link

Running on Custom Dataset #12

Closed Prudhvinik1 closed 5 years ago

Prudhvinik1 commented 5 years ago

Interested in this work, kindly provide necessary instructions and scripts for running Graph rcnn on custom datasets.

zc-alexfan commented 5 years ago

The author is still working on this repo's release.

jwyang commented 5 years ago

Hi, @Prudhvinik1 I have updated the readme for running the training and evaluation on scene graph generation models (vanilla, imp, frequency prior-based). The neural motif, graph r-cnn model are still on testing, and they will be here in very short time.