Spaces:
Runtime error
Runtime error
IDM-VTON-dedao-demo01
/
preprocess
/humanparsing
/mhp_extension
/detectron2
/tests
/test_model_zoo.py
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved | |
import logging | |
import unittest | |
from detectron2 import model_zoo | |
from detectron2.modeling import FPN, GeneralizedRCNN | |
logger = logging.getLogger(__name__) | |
class TestModelZoo(unittest.TestCase): | |
def test_get_returns_model(self): | |
model = model_zoo.get("Misc/scratch_mask_rcnn_R_50_FPN_3x_gn.yaml", trained=False) | |
self.assertIsInstance(model, GeneralizedRCNN) | |
self.assertIsInstance(model.backbone, FPN) | |
def test_get_invalid_model(self): | |
self.assertRaises(RuntimeError, model_zoo.get, "Invalid/config.yaml") | |
def test_get_url(self): | |
url = model_zoo.get_checkpoint_url("Misc/scratch_mask_rcnn_R_50_FPN_3x_gn.yaml") | |
self.assertEqual( | |
url, | |
"https://dl.fbaipublicfiles.com/detectron2/Misc/scratch_mask_rcnn_R_50_FPN_3x_gn/138602908/model_final_01ca85.pkl", # noqa | |
) | |
if __name__ == "__main__": | |
unittest.main() | |