From a0287daacf1868e5641ac35a5124c728e9f6953a Mon Sep 17 00:00:00 2001 From: gexy5 Date: Wed, 22 Jun 2022 19:17:03 +0800 Subject: [PATCH] modify supported model name Signed-off-by: gexy5 --- README.md | 1 - omnivore.py | 1 - 2 files changed, 2 deletions(-) diff --git a/README.md b/README.md index 6d63660..18e7ff1 100644 --- a/README.md +++ b/README.md @@ -72,7 +72,6 @@ model_name='omnivore_swinT', skip_preprocess=False, classmap=None, topk=5)*** - omnivore_swinB - omnivore_swinB_in21k - omnivore_swinL_in21k -- omnivore_swinB_epic ​ ***skip_preprocess***: *bool* diff --git a/omnivore.py b/omnivore.py index 5487a9b..464529f 100644 --- a/omnivore.py +++ b/omnivore.py @@ -28,7 +28,6 @@ class Omnivore(NNOperator): - omnivore_swinB - omnivore_swinB_in21k - omnivore_swinL_in21k - - omnivore_swinB_epic skip_preprocess (`str`): Flag to skip video transforms. predict (`bool`):