search

danfeiX / scene-graph-TF-release

"Scene Graph Generation by Iterative Message Passing" code repository
http://cs.stanford.edu/~danfei/scene-graph/
MIT License
425 stars 130 forks source link

Where do you separate Train/Val data? #31

Closed CongWeilin closed 5 years ago

CongWeilin commented 5 years ago

According to the paper, you mentioned 70% data for training and 30% for testing. However, I cannot find you data separation in your code. It seems to me that you are testing using training data. Please clarify if I miss something.

danfeiX commented 5 years ago

Split is encoded when creating the dataset at https://github.com/danfeiX/scene-graph-TF-release/blob/master/data_tools/vg_to_roidb.py#L586