ViewEvaluationNet icon indicating copy to clipboard operation
ViewEvaluationNet copied to clipboard

VEN for cvpr2018

View Evaluation Net

Requirements:

The VEN is implemented with Pytorch.

For viewing results using view_results.py, you will need matplotlib

Qucik Start

VEN is demonstrated in demo.py. Before running, update the following:

  1. Update the get_test_list() function to load images.
  2. Download the pre-trained mdoels from Google Drvie. Set the variable args.resume to point to it.

Train your own model

We will release the training code soon.