From d49215864114e4f9718cf113e2d5de05b6ab6d2e Mon Sep 17 00:00:00 2001 From: David Berard Date: Wed, 1 Nov 2023 20:03:07 -0700 Subject: [PATCH] detectron2: require coco2017-minimal data It seems like detectron2 requires coco2017-minimal data. It gets installed with vision-maskrcnn, but if you want to install requirements only for one model we need the installer scripts for the detectron models to download coco2017-minimal. --- torchbenchmark/util/framework/detectron2/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/torchbenchmark/util/framework/detectron2/__init__.py b/torchbenchmark/util/framework/detectron2/__init__.py index 50ce5900ff..954d671266 100644 --- a/torchbenchmark/util/framework/detectron2/__init__.py +++ b/torchbenchmark/util/framework/detectron2/__init__.py @@ -55,6 +55,7 @@ def remove_tools_directory(): def install_detectron2(model_name, model_dir): s3_utils.checkout_s3_data("INPUT_TARBALLS", "coco128.tar.gz", decompress=True) + s3_utils.checkout_s3_data("INPUT_TARBALLS", "coco2017-minimal.tar.gz", decompress=True) install_model_weights(model_name, model_dir) pip_install_requirements() remove_tools_directory()