Get the input image size for a given efficientnet model. Args: model_name (str): Name for efficientnet. Returns: Input image size (resolution).
(cls, model_name)
| 664 | |
| 665 | @classmethod |
| 666 | def get_image_size(cls, model_name): |
| 667 | """Get the input image size for a given efficientnet model. |
| 668 | Args: |
| 669 | model_name (str): Name for efficientnet. |
| 670 | Returns: |
| 671 | Input image size (resolution). |
| 672 | """ |
| 673 | cls._check_model_name_is_valid(model_name) |
| 674 | _, _, res, _ = efficientnet_params(model_name) |
| 675 | return res |
| 676 | |
| 677 | @classmethod |
| 678 | def _check_model_name_is_valid(cls, model_name): |
nothing calls this directly
no test coverage detected