reidddd commited on
Commit
bf103bf
·
1 Parent(s): 728451f

update import detectron2

Browse files
Files changed (2) hide show
  1. Dockerfile +15 -14
  2. requirements.txt +1 -0
Dockerfile CHANGED
@@ -1,3 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  FROM nvidia/cuda:11.1.1-cudnn8-devel-ubuntu18.04
2
  # use an older system (18.04) to avoid opencv incompatibility (issue#3524)
3
 
@@ -44,17 +58,4 @@ WORKDIR /home/user/detectron2_repo
44
  # python3 demo/demo.py \
45
  #--config-file configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml \
46
  #--input input.jpg --output outputs/ \
47
- #--opts MODEL.WEIGHTS detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl
48
- FROM python:3.9
49
-
50
- RUN useradd -m -u 1000 user
51
- USER user
52
- ENV PATH="/home/user/.local/bin:$PATH"
53
-
54
- WORKDIR /app
55
-
56
- COPY --chown=user ./requirements.txt requirements.txt
57
- RUN pip install --no-cache-dir --upgrade -r requirements.txt
58
-
59
- COPY --chown=user . /app
60
- CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
1
+
2
+ FROM python:3.9
3
+
4
+ RUN useradd -m -u 1000 user
5
+ USER user
6
+ ENV PATH="/home/user/.local/bin:$PATH"
7
+
8
+ WORKDIR /app
9
+
10
+ COPY --chown=user ./requirements.txt requirements.txt
11
+ RUN pip install --no-cache-dir --upgrade -r requirements.txt
12
+
13
+ COPY --chown=user . /app
14
+ CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
15
  FROM nvidia/cuda:11.1.1-cudnn8-devel-ubuntu18.04
16
  # use an older system (18.04) to avoid opencv incompatibility (issue#3524)
17
 
 
58
  # python3 demo/demo.py \
59
  #--config-file configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml \
60
  #--input input.jpg --output outputs/ \
61
+ #--opts MODEL.WEIGHTS detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl
 
 
 
 
 
 
 
 
 
 
 
 
 
requirements.txt CHANGED
@@ -12,3 +12,4 @@ Pillow
12
  gradio # Or streamlit, if using Streamlit
13
  opencv-python
14
  uvicorn
 
 
12
  gradio # Or streamlit, if using Streamlit
13
  opencv-python
14
  uvicorn
15
+ git+https://github.com/facebookresearch/detectron2.git