davidgillsjo/srw-net:a3b8d864 – Run with an API on Replicate

Version

You're looking at a specific version of this model. Jump to the model overview.

davidgillsjo /srw-net:a3b8d864

Playground API

Input

Run this model in Node.js with one line of code:

npx create-replicate --model=davidgillsjo/srw-net

or set up a project from scratch

Install Replicate’s Node.js client library:

npm install replicate

Set the REPLICATE_API_TOKEN environment variable:

export REPLICATE_API_TOKEN=<paste-your-token-here>

Find your API token in your account settings.

Import and set up the client:

import Replicate from "replicate";

const replicate = new Replicate({
  auth: process.env.REPLICATE_API_TOKEN,
});

Run davidgillsjo/srw-net using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

const output = await replicate.run(
  "davidgillsjo/srw-net:a3b8d86433a774113ebc48b1b853897dda957a0ff792c7bfb1d6b9781081994c",
  {
    input: {
      input_image: "https://replicate.delivery/mgxm/5bc4c9ed-d97e-4ae8-bb15-4b10614c6040/room.jpg",
      output_format: "Json",
      confidence_threshold: 0.9
    }
  }
);
console.log(output);

To learn more, take a look at the guide on getting started with Node.js.

Install Replicate’s Python client library:

pip install replicate

Set the REPLICATE_API_TOKEN environment variable:

export REPLICATE_API_TOKEN=<paste-your-token-here>

Find your API token in your account settings.

Import the client:

import replicate

Run davidgillsjo/srw-net using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

output = replicate.run(
    "davidgillsjo/srw-net:a3b8d86433a774113ebc48b1b853897dda957a0ff792c7bfb1d6b9781081994c",
    input={
        "input_image": "https://replicate.delivery/mgxm/5bc4c9ed-d97e-4ae8-bb15-4b10614c6040/room.jpg",
        "output_format": "Json",
        "confidence_threshold": 0.9
    }
)
print(output)

To learn more, take a look at the guide on getting started with Python.

Set the REPLICATE_API_TOKEN environment variable:

export REPLICATE_API_TOKEN=<paste-your-token-here>

Find your API token in your account settings.

Run davidgillsjo/srw-net using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

curl -s -X POST \
  -H "Authorization: Bearer $REPLICATE_API_TOKEN" \
  -H "Content-Type: application/json" \
  -H "Prefer: wait" \
  -d $'{
    "version": "a3b8d86433a774113ebc48b1b853897dda957a0ff792c7bfb1d6b9781081994c",
    "input": {
      "input_image": "https://replicate.delivery/mgxm/5bc4c9ed-d97e-4ae8-bb15-4b10614c6040/room.jpg",
      "output_format": "Json",
      "confidence_threshold": 0.9
    }
  }' \
  https://api.replicate.com/v1/predictions

To learn more, take a look at Replicate’s HTTP API reference docs.

You can run this model locally using Cog. First, install Cog:

brew install cog

If you don’t have Homebrew, there are other installation options available.

Run this to download the model and run it in your local environment:

cog predict r8.im/davidgillsjo/srw-net@sha256:a3b8d86433a774113ebc48b1b853897dda957a0ff792c7bfb1d6b9781081994c \
  -i 'input_image="https://replicate.delivery/mgxm/5bc4c9ed-d97e-4ae8-bb15-4b10614c6040/room.jpg"' \
  -i 'output_format="Json"' \
  -i 'confidence_threshold=0.9'

To learn more, take a look at the Cog documentation.

Run this to download the model and run it in your local environment:

docker run -d -p 5000:5000 --gpus=all r8.im/davidgillsjo/srw-net@sha256:a3b8d86433a774113ebc48b1b853897dda957a0ff792c7bfb1d6b9781081994c
curl -s -X POST \
  -H "Content-Type: application/json" \
  -d $'{
    "input": {
      "input_image": "https://replicate.delivery/mgxm/5bc4c9ed-d97e-4ae8-bb15-4b10614c6040/room.jpg",
      "output_format": "Json",
      "confidence_threshold": 0.9
    }
  }' \
  http://localhost:5000/predictions

To learn more, take a look at the Cog documentation.

Output

{ "Lines": [ { "type": "floor", "x1": 82.52734375, "y1": 1037.5133056640625, "x2": 922.1691284179688, "y2": 915.3350830078125 }, { "type": "wall", "x1": 82.52734375, "y1": 1037.5133056640625, "x2": 86.00208282470703, "y2": 220.39064025878906 }, { "type": "floor", "x1": 82.52734375, "y1": 1037.5133056640625, "x2": 0.019237995147705078, "y2": 1210.5810546875 }, { "type": "window", "x1": 2047.9981689453125, "y1": 90.39749145507812, "x2": 1542.742431640625, "y2": 257.729736328125 }, { "type": "window", "x1": 2047.9989013671875, "y1": 891.8460083007812, "x2": 1530.9671630859375, "y2": 841.6561889648438 }, { "type": "window", "x1": 475.5211486816406, "y1": 394.607421875, "x2": 477.1760559082031, "y2": 783.86328125 }, { "type": "window", "x1": 475.5211486816406, "y1": 394.607421875, "x2": 88.05802154541016, "y2": 337.3016662597656 }, { "type": "window", "x1": 477.1760559082031, "y1": 783.86328125, "x2": 72.5675048828125, "y2": 804.9552001953125 }, { "type": "floor", "x1": 2047.99365234375, "y1": 1317.217041015625, "x2": 922.1691284179688, "y2": 915.3350830078125 }, { "type": "floor", "x1": 2047.99365234375, "y1": 1317.217041015625, "x2": 1511.9517822265625, "y2": 1099.5526123046875 }, { "type": "window", "x1": 1542.742431640625, "y1": 257.729736328125, "x2": 1530.9671630859375, "y2": 841.6561889648438 }, { "type": "wall", "x1": 922.1929931640625, "y1": 360.14544677734375, "x2": 922.1691284179688, "y2": 915.3350830078125 }, { "type": "floor", "x1": 922.1691284179688, "y1": 915.3350830078125, "x2": 0.0810084342956543, "y2": 1040.91796875 } ], "Junctions": [ { "type": "proper", "x": 82.52734375, "y": 1037.5133056640625 }, { "type": "false", "x": 2047.9981689453125, "y": 90.39749145507812 }, { "type": "false", "x": 2047.9989013671875, "y": 891.8460083007812 }, { "type": "proper", "x": 475.5211486816406, "y": 394.607421875 }, { "type": "proper", "x": 477.1760559082031, "y": 783.86328125 }, { "type": "false", "x": 2047.99365234375, "y": 1317.217041015625 }, { "type": "proper", "x": 88.05802154541016, "y": 337.3016662597656 }, { "type": "proper", "x": 1542.742431640625, "y": 257.729736328125 }, { "type": "proper", "x": 922.1929931640625, "y": 360.14544677734375 }, { "type": "proper", "x": 922.1691284179688, "y": 915.3350830078125 }, { "type": "proper", "x": 1530.9671630859375, "y": 841.6561889648438 }, { "type": "proper", "x": 86.00208282470703, "y": 220.39064025878906 }, { "type": "proper", "x": 72.5675048828125, "y": 804.9552001953125 }, { "type": "false", "x": 0.019237995147705078, "y": 1210.5810546875 }, { "type": "false", "x": 0.0810084342956543, "y": 1040.91796875 }, { "type": "proper", "x": 1511.9517822265625, "y": 1099.5526123046875 } ] }

{
  "completed_at": "2022-06-25T19:27:24.614428Z",
  "created_at": "2022-06-25T19:25:50.806890Z",
  "data_removed": false,
  "error": null,
  "id": "r3x4plyrjzbezel3zf6ilvpdi4",
  "input": {
    "input_image": "https://replicate.delivery/mgxm/5bc4c9ed-d97e-4ae8-bb15-4b10614c6040/room.jpg",
    "output_format": "Json",
    "confidence_threshold": 0.9
  },
  "logs": "/root/.pyenv/versions/3.8.13/lib/python3.8/site-packages/torch/functional.py:445: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at  ../aten/src/ATen/native/TensorShape.cpp:2157.)\n  return _VF.meshgrid(tensors, **kwargs)  # type: ignore[attr-defined]",
  "metrics": {
    "predict_time": 1.481367,
    "total_time": 93.807538
  },
  "output": {
    "Json": "{\n    \"Lines\": [\n        {\n            \"type\": \"floor\",\n            \"x1\": 82.52734375,\n            \"y1\": 1037.5133056640625,\n            \"x2\": 922.1691284179688,\n            \"y2\": 915.3350830078125\n        },\n        {\n            \"type\": \"wall\",\n            \"x1\": 82.52734375,\n            \"y1\": 1037.5133056640625,\n            \"x2\": 86.00208282470703,\n            \"y2\": 220.39064025878906\n        },\n        {\n            \"type\": \"floor\",\n            \"x1\": 82.52734375,\n            \"y1\": 1037.5133056640625,\n            \"x2\": 0.019237995147705078,\n            \"y2\": 1210.5810546875\n        },\n        {\n            \"type\": \"window\",\n            \"x1\": 2047.9981689453125,\n            \"y1\": 90.39749145507812,\n            \"x2\": 1542.742431640625,\n            \"y2\": 257.729736328125\n        },\n        {\n            \"type\": \"window\",\n            \"x1\": 2047.9989013671875,\n            \"y1\": 891.8460083007812,\n            \"x2\": 1530.9671630859375,\n            \"y2\": 841.6561889648438\n        },\n        {\n            \"type\": \"window\",\n            \"x1\": 475.5211486816406,\n            \"y1\": 394.607421875,\n            \"x2\": 477.1760559082031,\n            \"y2\": 783.86328125\n        },\n        {\n            \"type\": \"window\",\n            \"x1\": 475.5211486816406,\n            \"y1\": 394.607421875,\n            \"x2\": 88.05802154541016,\n            \"y2\": 337.3016662597656\n        },\n        {\n            \"type\": \"window\",\n            \"x1\": 477.1760559082031,\n            \"y1\": 783.86328125,\n            \"x2\": 72.5675048828125,\n            \"y2\": 804.9552001953125\n        },\n        {\n            \"type\": \"floor\",\n            \"x1\": 2047.99365234375,\n            \"y1\": 1317.217041015625,\n            \"x2\": 922.1691284179688,\n            \"y2\": 915.3350830078125\n        },\n        {\n            \"type\": \"floor\",\n            \"x1\": 2047.99365234375,\n            \"y1\": 1317.217041015625,\n            \"x2\": 1511.9517822265625,\n            \"y2\": 1099.5526123046875\n        },\n        {\n            \"type\": \"window\",\n            \"x1\": 1542.742431640625,\n            \"y1\": 257.729736328125,\n            \"x2\": 1530.9671630859375,\n            \"y2\": 841.6561889648438\n        },\n        {\n            \"type\": \"wall\",\n            \"x1\": 922.1929931640625,\n            \"y1\": 360.14544677734375,\n            \"x2\": 922.1691284179688,\n            \"y2\": 915.3350830078125\n        },\n        {\n            \"type\": \"floor\",\n            \"x1\": 922.1691284179688,\n            \"y1\": 915.3350830078125,\n            \"x2\": 0.0810084342956543,\n            \"y2\": 1040.91796875\n        }\n    ],\n    \"Junctions\": [\n        {\n            \"type\": \"proper\",\n            \"x\": 82.52734375,\n            \"y\": 1037.5133056640625\n        },\n        {\n            \"type\": \"false\",\n            \"x\": 2047.9981689453125,\n            \"y\": 90.39749145507812\n        },\n        {\n            \"type\": \"false\",\n            \"x\": 2047.9989013671875,\n            \"y\": 891.8460083007812\n        },\n        {\n            \"type\": \"proper\",\n            \"x\": 475.5211486816406,\n            \"y\": 394.607421875\n        },\n        {\n            \"type\": \"proper\",\n            \"x\": 477.1760559082031,\n            \"y\": 783.86328125\n        },\n        {\n            \"type\": \"false\",\n            \"x\": 2047.99365234375,\n            \"y\": 1317.217041015625\n        },\n        {\n            \"type\": \"proper\",\n            \"x\": 88.05802154541016,\n            \"y\": 337.3016662597656\n        },\n        {\n            \"type\": \"proper\",\n            \"x\": 1542.742431640625,\n            \"y\": 257.729736328125\n        },\n        {\n            \"type\": \"proper\",\n            \"x\": 922.1929931640625,\n            \"y\": 360.14544677734375\n        },\n        {\n            \"type\": \"proper\",\n            \"x\": 922.1691284179688,\n            \"y\": 915.3350830078125\n        },\n        {\n            \"type\": \"proper\",\n            \"x\": 1530.9671630859375,\n            \"y\": 841.6561889648438\n        },\n        {\n            \"type\": \"proper\",\n            \"x\": 86.00208282470703,\n            \"y\": 220.39064025878906\n        },\n        {\n            \"type\": \"proper\",\n            \"x\": 72.5675048828125,\n            \"y\": 804.9552001953125\n        },\n        {\n            \"type\": \"false\",\n            \"x\": 0.019237995147705078,\n            \"y\": 1210.5810546875\n        },\n        {\n            \"type\": \"false\",\n            \"x\": 0.0810084342956543,\n            \"y\": 1040.91796875\n        },\n        {\n            \"type\": \"proper\",\n            \"x\": 1511.9517822265625,\n            \"y\": 1099.5526123046875\n        }\n    ]\n}"
  },
  "started_at": "2022-06-25T19:27:23.133061Z",
  "status": "succeeded",
  "urls": {
    "get": "https://api.replicate.com/v1/predictions/r3x4plyrjzbezel3zf6ilvpdi4",
    "cancel": "https://api.replicate.com/v1/predictions/r3x4plyrjzbezel3zf6ilvpdi4/cancel"
  },
  "version": "a3b8d86433a774113ebc48b1b853897dda957a0ff792c7bfb1d6b9781081994c"
}

Generated in

1.5 seconds

Tweak it Report