Open-sourcing the code for "CLIP as RNN: Segment Countless Visual Con…

…cepts without Training Endeavor". https://arxiv.org/abs/2312.07661 PiperOrigin-RevId: 600601383
mmrech · Jan 23, 2024 · 1d49f2c · 1d49f2c
1 parent 91cf0d6
commit 1d49f2c
Show file tree

Hide file tree

Showing 8,977 changed files with 10,342 additions and 8,970 deletions.
diff --git a/...rning_Linear_Thresholds_from_Label_Proportions/bag_size_3/large_margin_3-sized_LLP_LTF.py b/...rning_Linear_Thresholds_from_Label_Proportions/bag_size_3/large_margin_3-sized_LLP_LTF.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/...earning_Linear_Thresholds_from_Label_Proportions/bag_size_3/processing_results_3_sized.py b/...earning_Linear_Thresholds_from_Label_Proportions/bag_size_3/processing_results_3_sized.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/...rning_Linear_Thresholds_from_Label_Proportions/bag_size_3/small_margin_3-sized_LLP-LTF.py b/...rning_Linear_Thresholds_from_Label_Proportions/bag_size_3/small_margin_3-sized_LLP-LTF.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/...rning_Linear_Thresholds_from_Label_Proportions/bag_size_4/large_margin_4-sized_LLP_LTF.py b/...rning_Linear_Thresholds_from_Label_Proportions/bag_size_4/large_margin_4-sized_LLP_LTF.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/...earning_Linear_Thresholds_from_Label_Proportions/bag_size_4/processing_results_4_sized.py b/...earning_Linear_Thresholds_from_Label_Proportions/bag_size_4/processing_results_4_sized.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/...rning_Linear_Thresholds_from_Label_Proportions/bag_size_4/small_margin_4-sized_LLP-LTF.py b/...rning_Linear_Thresholds_from_Label_Proportions/bag_size_4/small_margin_4-sized_LLP-LTF.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/runnables/train_enc_dec.py b/COSTAR/runnables/train_enc_dec.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/runnables/train_gnet.py b/COSTAR/runnables/train_gnet.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/runnables/train_msm.py b/COSTAR/runnables/train_msm.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/runnables/train_multi.py b/COSTAR/runnables/train_multi.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/runnables/train_rep_est.py b/COSTAR/runnables/train_rep_est.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/runnables/train_rmsn.py b/COSTAR/runnables/train_rmsn.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/scripts_release/baselines.sh b/COSTAR/scripts_release/baselines.sh
@@ -1,4 +1,4 @@
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/scripts_release/costar.sh b/COSTAR/scripts_release/costar.sh
@@ -1,4 +1,4 @@
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/__init__.py b/COSTAR/src/__init__.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/data/__init__.py b/COSTAR/src/data/__init__.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/data/cancer_sim/__init__.py b/COSTAR/src/data/cancer_sim/__init__.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/data/cancer_sim/cancer_simulation.py b/COSTAR/src/data/cancer_sim/cancer_simulation.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/data/cancer_sim/dataset.py b/COSTAR/src/data/cancer_sim/dataset.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/data/dataset_collection.py b/COSTAR/src/data/dataset_collection.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/data/m5/__init__.py b/COSTAR/src/data/m5/__init__.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/data/m5/load_data.py b/COSTAR/src/data/m5/load_data.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/data/m5/real_dataset.py b/COSTAR/src/data/m5/real_dataset.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/data/mimic_iii/__init__.py b/COSTAR/src/data/mimic_iii/__init__.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/data/mimic_iii/load_data.py b/COSTAR/src/data/mimic_iii/load_data.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/data/mimic_iii/real_dataset.py b/COSTAR/src/data/mimic_iii/real_dataset.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/data/mimic_iii/semi_synthetic_dataset.py b/COSTAR/src/data/mimic_iii/semi_synthetic_dataset.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/data/mimic_iii/utils.py b/COSTAR/src/data/mimic_iii/utils.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/models/__init__.py b/COSTAR/src/models/__init__.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/models/crn.py b/COSTAR/src/models/crn.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/models/ct.py b/COSTAR/src/models/ct.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/models/edct.py b/COSTAR/src/models/edct.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/models/gnet.py b/COSTAR/src/models/gnet.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/models/msm.py b/COSTAR/src/models/msm.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/models/rep_est/COTrans_utils/__init__.py b/COSTAR/src/models/rep_est/COTrans_utils/__init__.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/models/rep_est/COTrans_utils/encoder.py b/COSTAR/src/models/rep_est/COTrans_utils/encoder.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/models/rep_est/CT_utils/__init__.py b/COSTAR/src/models/rep_est/CT_utils/__init__.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/models/rep_est/CT_utils/encoder.py b/COSTAR/src/models/rep_est/CT_utils/encoder.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/models/rep_est/__init__.py b/COSTAR/src/models/rep_est/__init__.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/models/rep_est/cotrans.py b/COSTAR/src/models/rep_est/cotrans.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/models/rep_est/ct.py b/COSTAR/src/models/rep_est/ct.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/models/rep_est/moco.py b/COSTAR/src/models/rep_est/moco.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/models/rep_est/rep_est.py b/COSTAR/src/models/rep_est/rep_est.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/models/rmsn.py b/COSTAR/src/models/rmsn.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/models/time_varying_model.py b/COSTAR/src/models/time_varying_model.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/models/utils.py b/COSTAR/src/models/utils.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/models/utils_lstm.py b/COSTAR/src/models/utils_lstm.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/models/utils_transformer.py b/COSTAR/src/models/utils_transformer.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/run/__init__.py b/COSTAR/src/run/__init__.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/src/run/utils.py b/COSTAR/src/run/utils.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/utils/__init__.py b/COSTAR/utils/__init__.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/COSTAR/utils/ckpt.py b/COSTAR/utils/ckpt.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/...trastive_Representations_for_Learning_from_Label_Proportions/code/criteo/autoint_utils.py b/...trastive_Representations_for_Learning_from_Label_Proportions/code/criteo/autoint_utils.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/...astive_Representations_for_Learning_from_Label_Proportions/code/criteo/batch_generator.py b/...astive_Representations_for_Learning_from_Label_Proportions/code/criteo/batch_generator.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/..._Contrastive_Representations_for_Learning_from_Label_Proportions/code/criteo/dllp_main.py b/..._Contrastive_Representations_for_Learning_from_Label_Proportions/code/criteo/dllp_main.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/...trastive_Representations_for_Learning_from_Label_Proportions/code/criteo/preprocessing.py b/...trastive_Representations_for_Learning_from_Label_Proportions/code/criteo/preprocessing.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/...stive_Representations_for_Learning_from_Label_Proportions/code/criteo/selfclr_llp_main.py b/...stive_Representations_for_Learning_from_Label_Proportions/code/criteo/selfclr_llp_main.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 The Google Research Authors.
+# Copyright 2024 The Google Research Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.