search

DavidGillsjo / SRW-Net

Semantic Room Wireframe Detection from a single perspective image
MIT License
72 stars 9 forks source link

Add Docker environment & web demo #5

Closed ariel415el closed 2 years ago

ariel415el commented 2 years ago

Hey @DavidGillsjo ! 👋

Good work on this Interesting project project !

This pull request makes your model available to run online on Replicate.com. I used an open source tool called Cog to build this and a byproduct is also another docker images with all the dependencies sorted out for this repo.

The live demo is available here: https://replicate.com/davidgillsjo/srw-net. Please click "Claim this model" and sign with your GitHub account so you can edit the page, and we'll feature it on our website and tweet about it too. When claimed, the model will also be automatically linked to the arXiv page (under 'Demos')!

I'm also here to help you if you want to edit or change anything you didn't liek in the demo (For example change the normalization of images before inference like we talked about).

In case you're wondering who I am, I'm from Replicate, where we're trying to make machine learning reproducible. We got frustrated that we couldn't run all the really interesting ML work being done. So, we're going round and looking for models we like making them even more accesible.

DavidGillsjo commented 2 years ago

Wow, that looks neat! Thanks for including the model.

Regarding the mean and std I'm not sure ImageNet would be representative either. I think an indoor dataset like ScanNet or Matterport would be better. You are of course welcome to try if you find the mean and std of either dataset. I'm not sure it would make enough of a difference, the best course would be to transfer learn on a Room Layout dataset based on ScanNet or Matterport.

I've now claimed the model and will merge the request. Thanks for all the work!

Could we include predictions on a couple of images from the Structured3D-dataset even if they are synthetic? I could send you a few or upload them myself if you let me know how I add examples.

DavidGillsjo commented 2 years ago

@ariel415el How do I change the badge from "Demo & Docker image" to "Live Demo" or similar?

ariel415el commented 2 years ago

Great @DavidGillsjo !

Now that you've claimed the model you can edit it on your own. You can delete unwanted examples and add ones of your choice by running the demo on an image and then clicking the "add example button"

About the badge, I think this one is fixed but you can change it whatever you want (emoji or other badges). just keep the link https://replicate.com/davidgillsjo/srw-net there..

Feel free to ask me more questions. I'm here to help you make this demo as best as it can be.