Spaces:
Runtime error
Runtime error
fifa-tryon-demo
/
Self-Correction-Human-Parsing-for-ACGPN
/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() | |