├── .DS_Store ├── .gitignore ├── LICENSE ├── Matrixfisher ├── ModelNetSo3 │ ├── ModelNetSo3.py │ ├── __init__.py │ └── geometric_utils.py ├── Pascal3D │ ├── Pascal3D.py │ ├── Pascal3D_all.py │ ├── Pascal3D_render.py │ ├── __init__.py │ ├── environment.yml │ ├── main.py │ └── parse_off.py ├── README.md ├── UPNA │ └── UPNA.py ├── __init__.py ├── configs │ ├── modelnet_config.json │ ├── pascal3d_VOC_as_train.json │ ├── pascal3d_no_augment.json │ ├── pascal3d_no_embedding.json │ ├── pascal3d_no_warp.json │ ├── pascal3d_normal.json │ ├── pascal3d_simple_baseline.json │ ├── pascal3d_synth.json │ └── upna_config.json ├── dataloader_utils.py ├── environment.yml ├── geometric_utils.py ├── logger.py ├── logger_metric.py ├── loss.py ├── main.py ├── np_math.py ├── parse_off.py ├── resnet.py ├── script_compute_mean_UPNA.py ├── script_dump_modelnet_errors.py ├── script_evaluate_approximation_accuracy.py ├── script_extract_data_from_tensorboard.py ├── script_test_loss.py ├── test_logger.py ├── test_logger_metric.py ├── test_loss.py ├── torch_math.py ├── torch_norm_factor.py ├── visualize_modelnet.py └── visualize_pascal3d.py ├── README.md ├── agent.py ├── config.py ├── dataset ├── .DS_Store ├── Pascal3D │ ├── .DS_Store │ ├── Pascal3D.py │ ├── Pascal3D_all.py │ ├── Pascal3D_render.py │ ├── __init__.py │ ├── environment.yml │ ├── main.py │ └── parse_off.py ├── __init__.py ├── dataloader_utils.py ├── dataset_modelnet.py ├── dataset_pascal.py ├── dataset_raw.py ├── dataset_symsol.py └── lib │ ├── Dataset_Base.py │ ├── __init__.py │ ├── common.py │ ├── db_type.py │ ├── imagedata_lmdb.py │ └── yaml_netconf.py ├── eval.py ├── eval_uncondition.py ├── flow ├── __init__.py ├── affineflow.py ├── condition.py ├── flow.py ├── mobiusflow.py ├── rottrans.py └── squeezetrans.py ├── settings ├── base.yml ├── modelnet_fisher.yml ├── modelnet_uni.yml ├── pascal_fisher.yml ├── pascal_uni.yml ├── raw.yml ├── symsol.yml └── symsol2.yml ├── train.py ├── train_uncondition.py └── utils ├── fisher.py ├── networks.py ├── sd.py ├── utils.py └── visualization.py /.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/.DS_Store -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/LICENSE -------------------------------------------------------------------------------- /Matrixfisher/ModelNetSo3/ModelNetSo3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/ModelNetSo3/ModelNetSo3.py -------------------------------------------------------------------------------- /Matrixfisher/ModelNetSo3/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Matrixfisher/ModelNetSo3/geometric_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/ModelNetSo3/geometric_utils.py -------------------------------------------------------------------------------- /Matrixfisher/Pascal3D/Pascal3D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/Pascal3D/Pascal3D.py -------------------------------------------------------------------------------- /Matrixfisher/Pascal3D/Pascal3D_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/Pascal3D/Pascal3D_all.py -------------------------------------------------------------------------------- /Matrixfisher/Pascal3D/Pascal3D_render.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/Pascal3D/Pascal3D_render.py -------------------------------------------------------------------------------- /Matrixfisher/Pascal3D/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Matrixfisher/Pascal3D/environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/Pascal3D/environment.yml -------------------------------------------------------------------------------- /Matrixfisher/Pascal3D/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/Pascal3D/main.py -------------------------------------------------------------------------------- /Matrixfisher/Pascal3D/parse_off.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/Pascal3D/parse_off.py -------------------------------------------------------------------------------- /Matrixfisher/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/README.md -------------------------------------------------------------------------------- /Matrixfisher/UPNA/UPNA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/UPNA/UPNA.py -------------------------------------------------------------------------------- /Matrixfisher/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Matrixfisher/configs/modelnet_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/configs/modelnet_config.json -------------------------------------------------------------------------------- /Matrixfisher/configs/pascal3d_VOC_as_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/configs/pascal3d_VOC_as_train.json -------------------------------------------------------------------------------- /Matrixfisher/configs/pascal3d_no_augment.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/configs/pascal3d_no_augment.json -------------------------------------------------------------------------------- /Matrixfisher/configs/pascal3d_no_embedding.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/configs/pascal3d_no_embedding.json -------------------------------------------------------------------------------- /Matrixfisher/configs/pascal3d_no_warp.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/configs/pascal3d_no_warp.json -------------------------------------------------------------------------------- /Matrixfisher/configs/pascal3d_normal.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/configs/pascal3d_normal.json -------------------------------------------------------------------------------- /Matrixfisher/configs/pascal3d_simple_baseline.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/configs/pascal3d_simple_baseline.json -------------------------------------------------------------------------------- /Matrixfisher/configs/pascal3d_synth.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/configs/pascal3d_synth.json -------------------------------------------------------------------------------- /Matrixfisher/configs/upna_config.json: -------------------------------------------------------------------------------- 1 | { 2 | "type": "upna" 3 | } 4 | 5 | -------------------------------------------------------------------------------- /Matrixfisher/dataloader_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/dataloader_utils.py -------------------------------------------------------------------------------- /Matrixfisher/environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/environment.yml -------------------------------------------------------------------------------- /Matrixfisher/geometric_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/geometric_utils.py -------------------------------------------------------------------------------- /Matrixfisher/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/logger.py -------------------------------------------------------------------------------- /Matrixfisher/logger_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/logger_metric.py -------------------------------------------------------------------------------- /Matrixfisher/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/loss.py -------------------------------------------------------------------------------- /Matrixfisher/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/main.py -------------------------------------------------------------------------------- /Matrixfisher/np_math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/np_math.py -------------------------------------------------------------------------------- /Matrixfisher/parse_off.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/parse_off.py -------------------------------------------------------------------------------- /Matrixfisher/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/resnet.py -------------------------------------------------------------------------------- /Matrixfisher/script_compute_mean_UPNA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/script_compute_mean_UPNA.py -------------------------------------------------------------------------------- /Matrixfisher/script_dump_modelnet_errors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/script_dump_modelnet_errors.py -------------------------------------------------------------------------------- /Matrixfisher/script_evaluate_approximation_accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/script_evaluate_approximation_accuracy.py -------------------------------------------------------------------------------- /Matrixfisher/script_extract_data_from_tensorboard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/script_extract_data_from_tensorboard.py -------------------------------------------------------------------------------- /Matrixfisher/script_test_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/script_test_loss.py -------------------------------------------------------------------------------- /Matrixfisher/test_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/test_logger.py -------------------------------------------------------------------------------- /Matrixfisher/test_logger_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/test_logger_metric.py -------------------------------------------------------------------------------- /Matrixfisher/test_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/test_loss.py -------------------------------------------------------------------------------- /Matrixfisher/torch_math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/torch_math.py -------------------------------------------------------------------------------- /Matrixfisher/torch_norm_factor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/torch_norm_factor.py -------------------------------------------------------------------------------- /Matrixfisher/visualize_modelnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/visualize_modelnet.py -------------------------------------------------------------------------------- /Matrixfisher/visualize_pascal3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/Matrixfisher/visualize_pascal3d.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/README.md -------------------------------------------------------------------------------- /agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/agent.py -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/config.py -------------------------------------------------------------------------------- /dataset/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/dataset/.DS_Store -------------------------------------------------------------------------------- /dataset/Pascal3D/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/dataset/Pascal3D/.DS_Store -------------------------------------------------------------------------------- /dataset/Pascal3D/Pascal3D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/dataset/Pascal3D/Pascal3D.py -------------------------------------------------------------------------------- /dataset/Pascal3D/Pascal3D_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/dataset/Pascal3D/Pascal3D_all.py -------------------------------------------------------------------------------- /dataset/Pascal3D/Pascal3D_render.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/dataset/Pascal3D/Pascal3D_render.py -------------------------------------------------------------------------------- /dataset/Pascal3D/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/Pascal3D/environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/dataset/Pascal3D/environment.yml -------------------------------------------------------------------------------- /dataset/Pascal3D/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/dataset/Pascal3D/main.py -------------------------------------------------------------------------------- /dataset/Pascal3D/parse_off.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/dataset/Pascal3D/parse_off.py -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/dataset/__init__.py -------------------------------------------------------------------------------- /dataset/dataloader_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/dataset/dataloader_utils.py -------------------------------------------------------------------------------- /dataset/dataset_modelnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/dataset/dataset_modelnet.py -------------------------------------------------------------------------------- /dataset/dataset_pascal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/dataset/dataset_pascal.py -------------------------------------------------------------------------------- /dataset/dataset_raw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/dataset/dataset_raw.py -------------------------------------------------------------------------------- /dataset/dataset_symsol.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/dataset/dataset_symsol.py -------------------------------------------------------------------------------- /dataset/lib/Dataset_Base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/dataset/lib/Dataset_Base.py -------------------------------------------------------------------------------- /dataset/lib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/lib/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/dataset/lib/common.py -------------------------------------------------------------------------------- /dataset/lib/db_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/dataset/lib/db_type.py -------------------------------------------------------------------------------- /dataset/lib/imagedata_lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/dataset/lib/imagedata_lmdb.py -------------------------------------------------------------------------------- /dataset/lib/yaml_netconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/dataset/lib/yaml_netconf.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/eval.py -------------------------------------------------------------------------------- /eval_uncondition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/eval_uncondition.py -------------------------------------------------------------------------------- /flow/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /flow/affineflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/flow/affineflow.py -------------------------------------------------------------------------------- /flow/condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/flow/condition.py -------------------------------------------------------------------------------- /flow/flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/flow/flow.py -------------------------------------------------------------------------------- /flow/mobiusflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/flow/mobiusflow.py -------------------------------------------------------------------------------- /flow/rottrans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/flow/rottrans.py -------------------------------------------------------------------------------- /flow/squeezetrans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/flow/squeezetrans.py -------------------------------------------------------------------------------- /settings/base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/settings/base.yml -------------------------------------------------------------------------------- /settings/modelnet_fisher.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/settings/modelnet_fisher.yml -------------------------------------------------------------------------------- /settings/modelnet_uni.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/settings/modelnet_uni.yml -------------------------------------------------------------------------------- /settings/pascal_fisher.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/settings/pascal_fisher.yml -------------------------------------------------------------------------------- /settings/pascal_uni.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/settings/pascal_uni.yml -------------------------------------------------------------------------------- /settings/raw.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/settings/raw.yml -------------------------------------------------------------------------------- /settings/symsol.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/settings/symsol.yml -------------------------------------------------------------------------------- /settings/symsol2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/settings/symsol2.yml -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/train.py -------------------------------------------------------------------------------- /train_uncondition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/train_uncondition.py -------------------------------------------------------------------------------- /utils/fisher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/utils/fisher.py -------------------------------------------------------------------------------- /utils/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/utils/networks.py -------------------------------------------------------------------------------- /utils/sd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/utils/sd.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/utils/utils.py -------------------------------------------------------------------------------- /utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/RotationNormFlow/HEAD/utils/visualization.py --------------------------------------------------------------------------------