search

SarahwXU / HiSup

MIT License
132 stars 20 forks source link

Visualisation of Inference #2

Open Ankit-Vohra opened 2 years ago

Ankit-Vohra commented 2 years ago

Hello Author, First of all want to comment your approach looks promising and I want to test it I'm facing a couple of issues

  1. How to store the inference visualisation?
  2. Can we convert the output polygon into ESRI Shapefile or GeoJSON?
  3. Is it possible to store the black and white mask?
SarahwXU commented 2 years ago

Hi, Thanks for your interest. For the question 1, you can look into the function “save_viz” from utils/visualizer.py. https://github.com/SarahwXU/HiSup/blob/c1a201be3ff7b0a7e51888dc8675e8f0f7d4142f/hisup/utils/visualizer.py#L44 For the question 2, it is sure that the output polygons could be converted into ESRI Shapefile or GeoJSON as long as you have the geographic coordinates. But the public AICrowd dataset does not provide that, so we do not give the code. For the question 3, the predicted mask is acquired here: https://github.com/SarahwXU/HiSup/blob/c1a201be3ff7b0a7e51888dc8675e8f0f7d4142f/hisup/detector.py#L121. You may set a breakpoint or add the save code.

Ankit-Vohra commented 2 years ago

Is there any way I can use my personal developed model and just try the image post processing work which you are doing to regularise the building edges

SarahwXU commented 2 years ago

First of all, I am not sure that I fully understand your meaning. A detailed description about the function of your personal developed model would help. Our work of HiSup emphasizes the interaction among the three representations (vertices, line segments, masks) and the joint training under all supervisions boosts the performance. From what I understand, you could add the supervisions from HiSup to your model, therefore, get a regular polygonal mapping.