wheel

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.whl filter=lfs diff=lfs merge=lfs -text

Dockerfile

@@ -8,13 +8,10 @@ WORKDIR /code
 COPY ./requirements.txt /code/requirements.txt
 COPY ./pre-requirements.txt /code/pre-requirements.txt
 COPY ./pytorch3d /code/pytorch3d
-COPY ./detectron2 /code/detectron2
 
+RUN pip install --no-cache-dir -r /code/pre-requirements.txt
 RUN pip install --no-cache-dir -r /code/requirements.txt
 
-RUN pip install --no-cache-dir ./pytorch3d
-RUN pip install --no-cache-dir ./detectron2
-
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
 

pre-requirements.txt

@@ -0,0 +1,8 @@
+ninja
+iopath
+fvcore
+
+--extra-index-url https://download.pytorch.org/whl/cpu
+torch==2.2.0 
+torchvision==0.17.0 
+torchaudio==2.2

pytorch3d/.gitignore

@@ -1,5 +1,4 @@
 build/
-dist/
 *.egg-info/
 **/__pycache__/
 *-checkpoint.ipynb

pytorch3d/dist/pytorch3d-0.7.8-cp311-cp311-manylinux1_x86_64.whl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1b815bf60f19498b5dcfaeb5eef66fa82c8f77d7b48616d7445729cd31e334b0
+size 878048

requirements.txt

@@ -4,11 +4,5 @@ numpy
 opencv-python
 pyransac3d
 
-ninja
-iopath
-fvcore
-
---extra-index-url https://download.pytorch.org/whl/cpu
-torch==2.2.0 
-torchvision==0.17.0 
-torchaudio==2.2.
+git+https://github.com/facebookresearch/detectron2.git
+pytorch3d/dist/pytorch3d-0.7.8-cp311-cp311-manylinux1_x86_64.whl