如何使用jquery自动将宽度设置为图像？

Question

我正在做一个模态照片库。为了使事情顺利进行，我需要将高度或宽度设置为100％取决于一个长度是否大于另一个长度。添加一个包含高度集的类正在工作，但是宽度有一个小问题。它不适用于此！帮助

这是jquery

    $(function() {
    if($('.modal-img').height() > $('.modal-img').width()) {
     $('.modal-img').addClass('set-height');
    }
      else if($('.modal-img').height() < $('.modal-img').width()) {
      $('.modal-img').addClass('set-width');
    }
    else {
       $('.modal-img').addClass('set-height');
       $('.modal-img').addClass('set-width');
    }
    });

你能帮忙吗？请。我不知道是不是因为我设置了这些图像的绝对位置，但我希望不是。