search

liamw96 / pytorch.room.layout

room layout estimation pytorch project
45 stars 10 forks source link

How to run the inference and visualization demo based on pretrained model #2

Closed SeaOfOcean closed 4 years ago

liamw96 commented 5 years ago

could you please elaborate your problems?

alex-kravets commented 5 years ago

I think it will be much convenient if you provide 'info.json' files separately from the zip archives. Maybe it was the case, I'm not sure :) But it was my very first idea when I decided to try to run inference on pre-trained model. BTW, can you please upload those json files? It is quite inconvenient to download at least 1.4G dataset to get single file) Thanks !

liamw96 commented 5 years ago

@alex-kravets Hi there! Actually for natural images, one can reuse the color statistics. For example, this one is just fine: https://github.com/fyu/drn/blob/master/datasets/cityscapes/info.json

You can also use the pytorch official info.json for ImageNet

run-me commented 5 years ago

Screenshot_2019-08-23_16-30-00 The model works like a charm, @liamw96 thanks for all your efforts and making this work open source. Working on vis methods will create a pull request when I am done working on it. Thanks again man! cheers!

liamw96 commented 5 years ago

@run-me thanks for your encouraging comments!

Chris20200910 commented 2 years ago

Hi, I hope get results like image But I only get results like image

So, what should I do?