RiceD2KLab · Jeffrey-Joan · Dec 14, 2024 · Dec 14, 2024 · Dec 14, 2024 · Dec 15, 2024
diff --git a/.DS_Store b/.DS_Store
diff --git a/.gitignore b/.gitignore
@@ -152,4 +152,4 @@ __pycache__/
 *.joblib
 *.pb
 *.ckpt
-*.pt
+*.ptdata/final_dataset/
diff --git a/README.md b/README.md
@@ -22,6 +22,7 @@ This project aims to develop a real-time segmentation algorithm for in-space ins
 - loguru==0.4.6
 - wandb==0.18.5
 - python-dotenv==1.0.1
+- diffusers==0.31.0
 
 ### Inference Dependencies
 - numpy==1.23.5

diff --git a/data/.DS_Store b/data/.DS_Store
diff --git a/data/FOV_full_data.svg b/data/FOV_full_data.svg
diff --git a/data/FOV_full_data_train.svg b/data/FOV_full_data_train.svg
-Original file line number
+Diff line change
@@ Expand Up / @@ -152,4 +152,4 @@ __pycache__/ @@
     *.joblib
     *.pb
     *.ckpt
-    *.pt
+    *.ptdata/final_dataset/