fidler-lab/polyrnn-pp-pytorch

This is the official PyTorch reimplementation of Polygon-RNN (CVPR 2018). This repository allows you to train new Polygon-RNN models, and run our demo tool on local machines. To get the code, please signup here. We will be using GitHub to keep track of issues with the code and to update on availability of newer versions (also available on website and through e-mail to signed up users). Note: Benchmarked forward pass speed for the tool (with 5 first points and 1 beam size) is 0.3 seconds per interaction on a TitanXp Note: Shared Encoder refers to sharing the Resnet between the graph network and the convLSTM network.