Skip to content
A vertex prediction model implementation based on https://arxiv.org/abs/1704.05548
Branch: master
Clone or download
Latest commit e346ef1 Sep 8, 2018
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
__pycache__
vertex_pred
.DS_Store
__init__.py
readme.md
requirements.txt

readme.md

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
You can’t perform that action at this time.