name: simplerecon
channels:
  - default
  - pytorch
  - conda-forge
dependencies:
  - clang=15.0.7
  - llvm-openmp=15.0.7
  - python=3.9.7
  - pytorch=1.10.0
  - torchvision=0.11.1
  - cudatoolkit=11.3
  - pytorch-lightning=1.5.4 # training utils
  - pillow # image op
  - tensorboard # logging
  - matplotlib # plotting
  - pip
  - pip:
    - kornia==0.6.7 # gradients
    - antialiased-cnns # anti aliased resnet
    - efficientnet_pytorch
    - timm # efficent
    - trimesh # mesh loading/storage, and mesh generation
    - transforms3d # for NeuralRecon's arkit
    - einops # batching one liners
    - moviepy # storing videos
    - pyrender # rendering meshes
    - open3d==0.14.1 # mesh fusion
    - scipy # transformations and a few others
    - protobuf<4.21.0 # lighting/tensorboard fix
    - setuptools==59.5.0 # fix for tensorboard
    - https://github.com/JamieWatson683/scikit-image/archive/single_mesh.zip # single mesh exporting for measure.marching_cubes
