search

Lydorn / Polygonization-by-Frame-Field-Learning

This repository contains the code for our fast polygonal building extraction from overhead images pipeline.
BSD 3-Clause "New" or "Revised" License
294 stars 70 forks source link

Polygonise mask on custom dataset #18

Open yeshwanth95 opened 3 years ago

yeshwanth95 commented 3 years ago

Hi @Lydorn ! I've been finding this repo really useful in my research and have a certain issue with one of the experiments I'm running using your method. I am trying to use your frame field learning approach on a custom version of the INRIA dataset where each image tile has been spit into a hundred 500x500 patches. When I try to run the polygonize_mask.py script on this dataset using one of your pretrained runs, I get the following error:

RuntimeError: The size of tensor a (1024) must match the size of tensor b (500) at non-singleton dimension 3

I am using the 'inria_dataset_osm_mask_only.unet16' run for this. Do you have any suggestions on how I could resolve this? Or would I need to train a different model from scratch to allow processing my modified patch size? Looking forward to your reply. Thank you.

vishwa15 commented 2 years ago

The size of the image must be large. I think atleast larger than 1024x1024. I tried with big TIF image file- it works !!