Swift：使用UIImage大小以编程方式调整uicollectionviewcell的高度

Question

我有集合视图，并以相同的宽度加载图像，但高度可变。我尝试使用contentView.systemLayoutSizeFitting进行修复，但无法正常工作。这是我的代码：

func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
    let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "cell", for: indexPath) as! CollectionViewCell
    let img = UIImage(named:array[indexPath.row])
    let imgWidth = img?.size.width
    let imgHeigh = img?.size.height
    let size = CGSize(width: imgWidth ?? 500, height: imgHeigh ?? 500)

    cell.myImgView.image = img
    cell.myImgView.backgroundColor = UIColor.blue
    cell.contentView.systemLayoutSizeFitting(size)
    return cell
}

任何人都知道如何根据图像的高度确定uicollectionviewcell的高度吗？

非常感谢您的帮助。