Skip to content

AidanRocke/vertex_prediction

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

TensorFlow implementation of vertex prediction model:

  1. The following repository includes a vertex prediction model that may be used to infer mappings from objects in images to the vertices of the polygon which bound the object of interest.
  2. Using a modified VGG-16 convolutional neural network that takes as input a 224x224x3 RGB image, inference is done in one forward pass that happens in two stages. First, an image representation is extracted from blocks of the VGG-16 model. Then a down-sampled variant of this image-representation is passed to a fully-connected layer which performs a global optimisation step.
  3. For more details concerning the reference paper, I would recommend reading 'Annotating Object Instances with a Polygon-RNN': https://arxiv.org/pdf/1704.05548.pdf as well as: https://arxiv.org/pdf/1803.09693.pdf

About

A vertex prediction model implementation based on https://arxiv.org/abs/1704.05548

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages