From 5fe9ac624b69512e7abf1f79f636b36e80d68c6b Mon Sep 17 00:00:00 2001 From: Albert Suarez Date: Tue, 8 Dec 2020 11:24:52 +0100 Subject: [PATCH] :memo: Updated underlaying model section --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index a8f1ccf..5f7b32e 100644 --- a/README.md +++ b/README.md @@ -113,7 +113,7 @@ _That's it_! You have ObjectCut running on port 80 routing traffic using _traefi ### Change underlying model -This project was built using [BASNet](https://github.com/NathanUA/BASNet) as the model for inferring the Salient Object Detection. However, in order to test other ones we added the support to select also [U^2-Net](https://github.com/NathanUA/U-2-Net), also implemented by [Xuebin Qin](https://github.com/NathanUA), in the Inference container specifying it as a environment variable called `MODEL`. You can do that setting your model name at [docker-compose.yml](docker-compose.yml): +This project was built using [BASNet](https://github.com/NathanUA/BASNet) as the model for inferring the Salient Object Detection. However, in order to test other ones we added the support to select also the different versions of [U^2-Net](https://github.com/NathanUA/U-2-Net) (`U2NET`, `U2NETP` and `U2NETPORTRAIT`), also implemented by [Xuebin Qin](https://github.com/NathanUA), in the Inference container specifying it as a environment variable called `MODEL`. You can do that setting your model name at [docker-compose.yml](docker-compose.yml): ```yaml inference: @@ -130,7 +130,7 @@ inference: - object_cut restart: always environment: - - MODEL=BASNet # Can also be `U2NET` + - MODEL=BASNet # Can also be `U2NET`, `U2NETP` or `U2NETPORTRAIT` ``` ### Integrations