feat: zoo.pointcloud_models [placeholder]

InterDigitalInc · Feb 2, 2024 · 5b97f67 · 5b97f67
1 parent 3697824
commit 5b97f67
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/compressai/zoo/__init__.py b/compressai/zoo/__init__.py
@@ -49,10 +49,18 @@
     "cheng2020-attn": cheng2020_attn,
 }
 
+# Not yet available.
+pointcloud_models = {
+    "hrtzxf2022-pcc-rec": None,
+    "sfu2023-pcc-rec-pointnet": None,
+    "sfu2024-pcc-rec-pointnet2-ssg": None,
+}
+
 video_models = {
     "ssf2020": ssf2020,
 }
 
 models = {}
 models.update(image_models)
+models.update(pointcloud_models)
 models.update(video_models)