文件名称：ebm_compositionality:纸张组成视觉生成和基于能量的模型推断的代码
文件大小：58KB
文件格式：ZIP
更新时间：2024-05-24 13:05:37
Python
进行代码可在找到关于合成的pytorch代码库。
安装先决条件
请通过运行以下命令安装所需的python软件包：
pip install -r requirements.txt
下载数据集
我们在Mujoco Scenes和CelebA数据集上进行实验。 要生成用于Mujoco Scenes数据集中的数据，请查看image_comb目录（您将需要适当地修改路径）并在其中运行相应的文件。 例如，要生成持续学习数据集，可以使用以下命令：
python image_comb/cube_continual.py
随时与我们联系以获取预先生成的Mujoco场景数据集
您可以在下载CelebA数据集
训练
使用以下命令训练模型：
python train.py --dataset=
【文件预览】：
ebm_compositionality-master
----celeba_combine.py(10KB)
----image_comb()
--------cube_color.py(10KB)
--------cube_continual.py(11KB)
--------cube.py(9KB)
--------cube_general.py(7KB)
--------multiview.py(6KB)
--------cube_joint.py(8KB)
--------cube_cond.py(10KB)
----models.py(14KB)
----utils.py(30KB)
----train.py(40KB)
----filters.py(466B)
----imagenet_demo.py(3KB)
----requirements.txt(289B)
----data_vis.py(5KB)
----ebm_sandbox.py(44KB)
----custom_adam.py(10KB)
----README.md(2KB)
----data.py(12KB)