|
|
|
|
|
|
|
|
RUN apt update && apt install -y zip htop screen libgl1-mesa-glx |
|
|
RUN apt update && apt install -y zip htop screen libgl1-mesa-glx |
|
|
|
|
|
|
|
|
# Install python dependencies |
|
|
# Install python dependencies |
|
|
RUN python -m pip install --upgrade pip |
|
|
|
|
|
COPY requirements.txt . |
|
|
COPY requirements.txt . |
|
|
|
|
|
RUN python -m pip install --upgrade pip |
|
|
RUN pip install --no-cache -r requirements.txt gsutil notebook |
|
|
RUN pip install --no-cache -r requirements.txt gsutil notebook |
|
|
|
|
|
|
|
|
# Create working directory |
|
|
# Create working directory |
|
|
|
|
|
|
|
|
# Copy contents |
|
|
# Copy contents |
|
|
COPY . /usr/src/app |
|
|
COPY . /usr/src/app |
|
|
|
|
|
|
|
|
# Copy weights |
|
|
|
|
|
#RUN python3 -c "from models import *; \ |
|
|
|
|
|
#attempt_download('weights/yolov5s.pt'); \ |
|
|
|
|
|
#attempt_download('weights/yolov5m.pt'); \ |
|
|
|
|
|
#attempt_download('weights/yolov5l.pt')" |
|
|
|
|
|
|
|
|
# Set environment variables |
|
|
|
|
|
ENV HOME=/usr/src/app |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# --------------------------------------------------- Extras Below --------------------------------------------------- |
|
|
# --------------------------------------------------- Extras Below --------------------------------------------------- |