Last active
October 11, 2021 11:40
-
-
Save tothandras/223364a86c05da952fe43967fb14f619 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
FROM nvcr.io/nvidia/tensorflow:20.01-tf2-py3 | |
RUN apt-get update && \ | |
apt-get install -y libsm6 libxext6 libxrender-dev && \ | |
pip install opencv-python | |
COPY . . | |
# RUN echo "Downloading weights..." && \ | |
# wget -O weights/yolov3.weights https://pjreddie.com/media/files/yolov3.weights && \ | |
# wget -O weights/yolov3-tiny.weights https://pjreddie.com/media/files/yolov3-tiny.weights && \ | |
# wget -O weights/yolov3-spp.weights https://pjreddie.com/media/files/yolov3-spp.weights && \ | |
# wget -O weights/darknet53.conv.74.weights https://pjreddie.com/media/files/darknet53.conv.74 && \ | |
# wget -O weights/darknet19_448.conv.23.weights https://pjreddie.com/media/files/darknet19_448.conv.23 && \ | |
# wget -O weights/yolov2.weights http://pjreddie.com/media/files/yolo.weights && \ | |
# wget -O weights/yolov2-voc.weights http://pjreddie.com/media/files/yolo-voc.weights | |
# RUN echo "Converting Darknet YOLO to Keras models..." && \ | |
# python tools/convert.py cfg/yolov3.cfg weights/yolov3.weights weights/yolov3.h5 && \ | |
# python tools/convert.py cfg/yolov3-tiny.cfg weights/yolov3-tiny.weights weights/yolov3-tiny.h5 && \ | |
# python tools/convert.py cfg/yolov3-spp.cfg weights/yolov3-spp.weights weights/yolov3-spp.h5 && \ | |
# python tools/convert.py cfg/yolov2.cfg weights/yolov2.weights weights/yolov2.h5 && \ | |
# python tools/convert.py cfg/yolov2-voc.cfg weights/yolov2-voc.weights weights/yolov2-voc.h5 && \ | |
# python tools/convert.py cfg/darknet53.cfg weights/darknet53.conv.74.weights weights/darknet53.h5 && \ | |
# python tools/convert.py cfg/darknet19_448_body.cfg weights/darknet19_448.conv.23.weights weights/darknet19.h5 | |
RUN wget https://github.com/david8862/keras-YOLOv3-model-set/releases/download/v1.0.1/yolo3_nano_weights_416_voc.tar.gz && \ | |
tar zxvf yolo3_nano_weights_416_voc.tar.gz && \ | |
mv yolo3_nano_weights_416_voc.h5 ./weights/yolo3_nano_weights_416_voc.h5 && \ | |
rm yolo3_nano_weights_416_voc.tar.gz | |
RUN mkdir -p data/PascalVOC && \ | |
wget -O data/PascalVOC/VOCtrainval_06-Nov-2007.tar http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar && \ | |
wget -O data/PascalVOC/VOCtest_06-Nov-2007.tar http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar && \ | |
wget -O data/PascalVOC/VOCtrainval_11-May-2012.tar http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar && \ | |
wget -O data/PascalVOC/VOC2012test.tar http://pjreddie.com/media/files/VOC2012test.tar && \ | |
cd data/PascalVOC && \ | |
tar xf VOCtest_06-Nov-2007.tar && \ | |
tar xf VOCtrainval_06-Nov-2007.tar && \ | |
tar xf VOCtrainval_11-May-2012.tar && \ | |
cd ../.. | |
RUN python tools/voc_annotation.py --dataset_path=/workspace/data/PascalVOC/VOCdevkit/ --output_path=/workspace/data/PascalVOC | |
RUN cd data/PascalVOC && \ | |
touch trainval.txt && \ | |
cat 2007_train.txt >> trainval.txt && cat 2007_train.txt >> trainval.txt && cat 2012_train.txt >> trainval.txt && cat 2012_val.txt >> trainval.txt && \ | |
cp -rf trainval.txt 2007_test.txt ../.. && \ | |
cd ../.. | |
RUN pip install -r requirements.txt |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment