improve simplicity, allow robot hand seg

Files changed (11) hide show

.gitignore CHANGED Viewed

@@ -3,4 +3,7 @@ sbatch/err/
 sbatch/out/
 __pycache__/
 diffusion/__pycache__/
-*.pyc

 sbatch/out/
 __pycache__/
 diffusion/__pycache__/
+*.pyc
+DINO_EMA_11M_b50_lr1e-5_epoch6_step320k.ckpt
+sam_vit_h_4b8939.pth
+vae-ft-mse-840000-ema-pruned.ckpt

app.py CHANGED Viewed

The diff for this file is too large to render. See raw diff

diffusion/__pycache__/__init__.cpython-38.pyc CHANGED Viewed

Binary files a/diffusion/__pycache__/__init__.cpython-38.pyc and b/diffusion/__pycache__/__init__.cpython-38.pyc differ

diffusion/__pycache__/diffusion_utils.cpython-38.pyc CHANGED Viewed

Binary files a/diffusion/__pycache__/diffusion_utils.cpython-38.pyc and b/diffusion/__pycache__/diffusion_utils.cpython-38.pyc differ

diffusion/__pycache__/gaussian_diffusion.cpython-38.pyc CHANGED Viewed

Binary files a/diffusion/__pycache__/gaussian_diffusion.cpython-38.pyc and b/diffusion/__pycache__/gaussian_diffusion.cpython-38.pyc differ

diffusion/__pycache__/respace.cpython-38.pyc CHANGED Viewed

Binary files a/diffusion/__pycache__/respace.cpython-38.pyc and b/diffusion/__pycache__/respace.cpython-38.pyc differ

diffusion/__pycache__/scheduler.cpython-38.pyc CHANGED Viewed

Binary files a/diffusion/__pycache__/scheduler.cpython-38.pyc and b/diffusion/__pycache__/scheduler.cpython-38.pyc differ

no_hands_open.jpeg ADDED Viewed

sample_images/sample10.jpg CHANGED Viewed

sample_images/sample3.jpg CHANGED Viewed

sample_images/sample4.jpg CHANGED Viewed