Полноэкранный UIImage в Swift 4

Question

У меня есть UICollectionView с ячейками, которые содержат UIImages.Вот мой метод didSelectIndex для захвата изображения и его полноэкранного просмотра.

override func collectionView(_ collectionView: UICollectionView, didSelectItemAt indexPath: IndexPath) {
    let media = self.media[indexPath.row]
    guard let photoURL = URL(string: media.photoURL) else { return }

    // Get the image from the cache
    if let image = mediaCache.object(forKey: photoURL.absoluteString as AnyObject) as? UIImage {
        let imageView = UIImageView(image: image)
        imageView.frame = UIScreen.main.bounds
        imageView.backgroundColor = .black
        imageView.contentMode = .scaleAspectFill
        imageView.clipsToBounds = true
        imageView.isUserInteractionEnabled = true

        self.navigationController?.isNavigationBarHidden = true
        self.tabBarController?.tabBar.isHidden = true

        let dismissTap = UITapGestureRecognizer(target: self, action: #selector(dismissFullscreenImage))
        imageView.addGestureRecognizer(dismissTap)

        self.view.addSubview(imageView)
    }
}

Однако изменение режима содержимого с scaleAspectFill на scaleAspectFit не решает проблему.Изображение не помещается внутри фрейма UIImageView.