KaihuaTang / Scene-Graph-Benchmark.pytorch

A new codebase for popular Scene Graph Generation methods (2020). Visualization & Scene Graph Extraction on custom images/datasets are provided. It's also a PyTorch implementation of paper “Unbiased Scene Graph Generation from Biased Training CVPR 2020”
MIT License
1.03k stars 228 forks source link

Training Recommendation for Image Retrieval #164

Closed rafiberlin closed 1 year ago

rafiberlin commented 2 years ago

This is just a small amendment in the description. Training the Image Retrieval model for 9 epochs delivers better results.