YOLOv5 Forward Compatibility Update (#1569)
* YOLOv5 forward compatibility update * add data dir * ci test yolov3 * update build_targets() * update build_targets() * update build_targets() * update yolov3-spp.yaml * add yolov3-tiny.yaml * add yolov3-tiny.yaml * Update yolov3-tiny.yaml * thop bug fix * Detection() device bug fix * Use torchvision.ops.nms() * Remove redundant download mirror * CI tests with yolov3-tiny * Update README.md * Synch train and test iou_thresh * update requirements.txt * Cat apriori autolabels * Confusion matrix * Autosplit * Autosplit * Update README.md * AP no plot * Update caching * Update caching * Caching bug fix * --image-weights bug fix * datasets bug fix * mosaic plots bug fix * plot_study * boxes.max() * boxes.max() * boxes.max() * boxes.max() * boxes.max() * boxes.max() * update * Update README * Update README * Update README.md * Update README.md * results png * Update README * Targets scaling bug fix * update plot_study * update plot_study * update plot_study * update plot_study * Targets scaling bug fix * Finish Readme.md * Finish Readme.md * Finish Readme.md * Update README.md * Creado con Colaboratory
This commit is contained in:
Glenn Jocher
GitHub
+26
-36
@@ -1,32 +1,11 @@
|
||||
# Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
|
||||
FROM nvcr.io/nvidia/pytorch:20.03-py3
|
||||
FROM nvcr.io/nvidia/pytorch:20.10-py3
|
||||
|
||||
# Install dependencies (pip or conda)
|
||||
RUN pip install -U gsutil
|
||||
# RUN pip install -U -r requirements.txt
|
||||
# RUN conda update -n base -c defaults conda
|
||||
# RUN conda install -y -c anaconda future numpy opencv matplotlib tqdm pillow
|
||||
# RUN conda install -y -c conda-forge scikit-image tensorboard pycocotools
|
||||
|
||||
## Install OpenCV with Gstreamer support
|
||||
#WORKDIR /usr/src
|
||||
#RUN pip uninstall -y opencv-python
|
||||
#RUN apt-get update
|
||||
#RUN apt-get install -y gstreamer1.0-tools gstreamer1.0-python3-dbg-plugin-loader libgstreamer1.0-dev libgstreamer-plugins-base1.0-dev
|
||||
#RUN git clone https://github.com/opencv/opencv.git && cd opencv && git checkout 4.1.1 && mkdir build
|
||||
#RUN git clone https://github.com/opencv/opencv_contrib.git && cd opencv_contrib && git checkout 4.1.1
|
||||
#RUN cd opencv/build && cmake ../ \
|
||||
# -D OPENCV_EXTRA_MODULES_PATH=../../opencv_contrib/modules \
|
||||
# -D BUILD_OPENCV_PYTHON3=ON \
|
||||
# -D PYTHON3_EXECUTABLE=/opt/conda/bin/python \
|
||||
# -D PYTHON3_INCLUDE_PATH=/opt/conda/include/python3.6m \
|
||||
# -D PYTHON3_LIBRARIES=/opt/conda/lib/python3.6/site-packages \
|
||||
# -D WITH_GSTREAMER=ON \
|
||||
# -D WITH_FFMPEG=OFF \
|
||||
# && make && make install && ldconfig
|
||||
#RUN cd /usr/local/lib/python3.6/site-packages/cv2/python-3.6/ && mv cv2.cpython-36m-x86_64-linux-gnu.so cv2.so
|
||||
#RUN cd /opt/conda/lib/python3.6/site-packages/ && ln -s /usr/local/lib/python3.6/site-packages/cv2/python-3.6/cv2.so cv2.so
|
||||
#RUN python3 -c "import cv2; print(cv2.getBuildInformation())"
|
||||
# Install dependencies
|
||||
RUN pip install --upgrade pip
|
||||
# COPY requirements.txt .
|
||||
# RUN pip install -r requirements.txt
|
||||
RUN pip install gsutil
|
||||
|
||||
# Create working directory
|
||||
RUN mkdir -p /usr/src/app
|
||||
@@ -38,25 +17,36 @@ COPY . /usr/src/app
|
||||
# Copy weights
|
||||
#RUN python3 -c "from models import *; \
|
||||
#attempt_download('weights/yolov3.pt'); \
|
||||
#attempt_download('weights/yolov3-spp.pt')"
|
||||
#attempt_download('weights/yolov3-spp.pt'); \
|
||||
#attempt_download('weights/yolov3-tiny.pt')"
|
||||
|
||||
|
||||
# --------------------------------------------------- Extras Below ---------------------------------------------------
|
||||
|
||||
# Build and Push
|
||||
# t=ultralytics/yolov3:v0 && sudo docker build -t $t . && sudo docker push $t
|
||||
# t=ultralytics/yolov3:latest && sudo docker build -t $t . && sudo docker push $t
|
||||
# for v in {300..303}; do t=ultralytics/coco:v$v && sudo docker build -t $t . && sudo docker push $t; done
|
||||
|
||||
# Run
|
||||
# t=ultralytics/yolov3:v0 && sudo docker pull $t && sudo docker run -it --gpus all --ipc=host $t bash
|
||||
# Pull and Run
|
||||
# t=ultralytics/yolov3:latest && sudo docker pull $t && sudo docker run -it --ipc=host $t
|
||||
|
||||
# Pull and Run with local directory access
|
||||
# t=ultralytics/yolov3:v0 && sudo docker pull $t && sudo docker run -it --gpus all --ipc=host -v "$(pwd)"/coco:/usr/src/coco $t bash
|
||||
# t=ultralytics/yolov3:latest && sudo docker pull $t && sudo docker run -it --ipc=host --gpus all -v "$(pwd)"/coco:/usr/src/coco $t
|
||||
|
||||
# Kill all
|
||||
# sudo docker kill "$(sudo docker ps -q)"
|
||||
# sudo docker kill $(sudo docker ps -q)
|
||||
|
||||
# Kill all image-based
|
||||
# sudo docker kill $(sudo docker ps -a -q --filter ancestor=ultralytics/yolov3:v0)
|
||||
# sudo docker kill $(sudo docker ps -a -q --filter ancestor=ultralytics/yolov3:latest)
|
||||
|
||||
# Run bash for loop
|
||||
# sudo docker run --gpus all --ipc=host ultralytics/yolov3:v0 while true; do python3 train.py --evolve; done
|
||||
# Bash into running container
|
||||
# sudo docker container exec -it ba65811811ab bash
|
||||
|
||||
# Bash into stopped container
|
||||
# sudo docker commit 092b16b25c5b usr/resume && sudo docker run -it --gpus all --ipc=host -v "$(pwd)"/coco:/usr/src/coco --entrypoint=sh usr/resume
|
||||
|
||||
# Send weights to GCP
|
||||
# python -c "from utils.general import *; strip_optimizer('runs/train/exp0_*/weights/best.pt', 'tmp.pt')" && gsutil cp tmp.pt gs://*.pt
|
||||
|
||||
# Clean up
|
||||
# docker system prune -a --volumes
|
||||
|
||||
Reference in New Issue
Block a user