bmshj2018-factorized-relu model urls missing in zoo

[Mateen Ulhaq]

Bug

bmshj2018_factorized_relu loads "bmshj2018-factorized" (GDN):

CompressAI/compressai/zoo/image.py

Lines 334 to 336 in a4ae2ee

 return _load_model( 
     "bmshj2018-factorized", metric, quality, pretrained, progress, **kwargs 
 ) 

Furthermore, the model URLs for Amazon AWS are missing:

CompressAI/compressai/zoo/image.py

Lines 65 to 109 in a4ae2ee

 model_urls = { 
     "bmshj2018-factorized": { 
         "mse": { 
             1: f"{root_url}/bmshj2018-factorized-prior-1-446d5c7f.pth.tar", 
             2: f"{root_url}/bmshj2018-factorized-prior-2-87279a02.pth.tar", 
             3: f"{root_url}/bmshj2018-factorized-prior-3-5c6f152b.pth.tar", 
             4: f"{root_url}/bmshj2018-factorized-prior-4-1ed4405a.pth.tar", 
             5: f"{root_url}/bmshj2018-factorized-prior-5-866ba797.pth.tar", 
             6: f"{root_url}/bmshj2018-factorized-prior-6-9b02ea3a.pth.tar", 
             7: f"{root_url}/bmshj2018-factorized-prior-7-6dfd6734.pth.tar", 
             8: f"{root_url}/bmshj2018-factorized-prior-8-5232faa3.pth.tar", 
         }, 
         "ms-ssim": { 
             1: f"{root_url}/bmshj2018-factorized-ms-ssim-1-9781d705.pth.tar", 
             2: f"{root_url}/bmshj2018-factorized-ms-ssim-2-4a584386.pth.tar", 
             3: f"{root_url}/bmshj2018-factorized-ms-ssim-3-5352f123.pth.tar", 
             4: f"{root_url}/bmshj2018-factorized-ms-ssim-4-4f91b847.pth.tar", 
             5: f"{root_url}/bmshj2018-factorized-ms-ssim-5-b3a88897.pth.tar", 
             6: f"{root_url}/bmshj2018-factorized-ms-ssim-6-ee028763.pth.tar", 
             7: f"{root_url}/bmshj2018-factorized-ms-ssim-7-8c265a29.pth.tar", 
             8: f"{root_url}/bmshj2018-factorized-ms-ssim-8-8811bd14.pth.tar", 
         }, 
     }, 
     "bmshj2018-hyperprior": { 
         "mse": { 
             1: f"{root_url}/bmshj2018-hyperprior-1-7eb97409.pth.tar", 
             2: f"{root_url}/bmshj2018-hyperprior-2-93677231.pth.tar", 
             3: f"{root_url}/bmshj2018-hyperprior-3-6d87be32.pth.tar", 
             4: f"{root_url}/bmshj2018-hyperprior-4-de1b779c.pth.tar", 
             5: f"{root_url}/bmshj2018-hyperprior-5-f8b614e1.pth.tar", 
             6: f"{root_url}/bmshj2018-hyperprior-6-1ab9c41e.pth.tar", 
             7: f"{root_url}/bmshj2018-hyperprior-7-3804dcbd.pth.tar", 
             8: f"{root_url}/bmshj2018-hyperprior-8-a583f0cf.pth.tar", 
         }, 
         "ms-ssim": { 
             1: f"{root_url}/bmshj2018-hyperprior-ms-ssim-1-5cf249be.pth.tar", 
             2: f"{root_url}/bmshj2018-hyperprior-ms-ssim-2-1ff60d1f.pth.tar", 
             3: f"{root_url}/bmshj2018-hyperprior-ms-ssim-3-92dd7878.pth.tar", 
             4: f"{root_url}/bmshj2018-hyperprior-ms-ssim-4-4377354e.pth.tar", 
             5: f"{root_url}/bmshj2018-hyperprior-ms-ssim-5-c34afc8d.pth.tar", 
             6: f"{root_url}/bmshj2018-hyperprior-ms-ssim-6-3a6d8229.pth.tar", 
             7: f"{root_url}/bmshj2018-hyperprior-ms-ssim-7-8747d3bc.pth.tar", 
             8: f"{root_url}/bmshj2018-hyperprior-ms-ssim-8-cc15b5f3.pth.tar", 
         }, 
     },