collection of Dockerfile.
- cuda10.1-cudnn7-tenorrt6.0.1.5
- Yolov4-trt-deployment
- YOLOv4-darknet:
train/detect- For YOLOv4-darknet model training
- For YOLOv4-darknet model deploy
- tfod-train-1.15.2
- TensorFlow1.15.2 object detection training image.
- ...
from AlexeyAB/darknet
Thanks for the nice repo. from jkjung-avt, where almost all
*.pycodes borrown from.
default starting the container effctively behave as
docker run -itd -p 11221:5000 --name yolotrt512 --gpus '"device=5"' yolov4/deploy:trt -m yolov4-512You can simply change the CMD in the CLI as:
docker run -itd -p 11221:5000 --name yolotrt512 --gpus '"device=5"' yolov4/deploy:trt -m yolov4-416or using yolov4-608.
file structure:
├── deploy
│ ├── Dockerfile
│ └── deploy-standalone
│ ├── deploy_flaskAPI.py
│ ├── model
│ │ ├── libyolo_layer.so
│ │ ├── yolov4-416.trt
│ │ ├── yolov4-512.trt
│ │ └── yolov4-608.trt
│ ├── pets.jpeg
│ ├── this_is_a_test.jpg
│ ├── thisis_test_.jpg
│ ├── trt_yolo.py
│ └── utils
│ ├── visualization.py
│ ├── yolo_class.py
│ ├── yolo_classes.py
│ └── yolo_with_plugins.pybuild the image like (see veri.md):
docker build -t tfod:1.15.2 -f Dockerfile .other tfod details, please refer to tensorflow/models
when finish the image building process, test the installation in a container,
# Test the installation.
python object_detection/builders/model_builder_tf1_test.pyroot@246c6b10349f:/home/tensorflow/models/research# python object_detection/builders/model_builder_tf1_test.py
Running tests under Python 3.6.9: /usr/local/bin/python
[ RUN ] ModelBuilderTF1Test.test_create_context_rcnn_from_config_with_params(True)
[ OK ] ModelBuilderTF1Test.test_create_context_rcnn_from_config_with_params(True)
[ RUN ] ModelBuilderTF1Test.test_create_context_rcnn_from_config_with_params(False)
[ OK ] ModelBuilderTF1Test.test_create_context_rcnn_from_config_with_params(False)
[ RUN ] ModelBuilderTF1Test.test_create_experimental_model
[ OK ] ModelBuilderTF1Test.test_create_experimental_model
[ RUN ] ModelBuilderTF1Test.test_create_faster_rcnn_from_config_with_crop_feature(True)
[ OK ] ModelBuilderTF1Test.test_create_faster_rcnn_from_config_with_crop_feature(True)
[ RUN ] ModelBuilderTF1Test.test_create_faster_rcnn_from_config_with_crop_feature(False)
[ OK ] ModelBuilderTF1Test.test_create_faster_rcnn_from_config_with_crop_feature(False)
[ RUN ] ModelBuilderTF1Test.test_create_faster_rcnn_model_from_config_with_example_miner
[ OK ] ModelBuilderTF1Test.test_create_faster_rcnn_model_from_config_with_example_miner
[ RUN ] ModelBuilderTF1Test.test_create_faster_rcnn_models_from_config_faster_rcnn_with_matmul
[ OK ] ModelBuilderTF1Test.test_create_faster_rcnn_models_from_config_faster_rcnn_with_matmul
[ RUN ] ModelBuilderTF1Test.test_create_faster_rcnn_models_from_config_faster_rcnn_without_matmul
[ OK ] ModelBuilderTF1Test.test_create_faster_rcnn_models_from_config_faster_rcnn_without_matmul
[ RUN ] ModelBuilderTF1Test.test_create_faster_rcnn_models_from_config_mask_rcnn_with_matmul
[ OK ] ModelBuilderTF1Test.test_create_faster_rcnn_models_from_config_mask_rcnn_with_matmul
[ RUN ] ModelBuilderTF1Test.test_create_faster_rcnn_models_from_config_mask_rcnn_without_matmul
[ OK ] ModelBuilderTF1Test.test_create_faster_rcnn_models_from_config_mask_rcnn_without_matmul
[ RUN ] ModelBuilderTF1Test.test_create_rfcn_model_from_config
[ OK ] ModelBuilderTF1Test.test_create_rfcn_model_from_config
[ RUN ] ModelBuilderTF1Test.test_create_ssd_fpn_model_from_config
[ OK ] ModelBuilderTF1Test.test_create_ssd_fpn_model_from_config
[ RUN ] ModelBuilderTF1Test.test_create_ssd_models_from_config
[ OK ] ModelBuilderTF1Test.test_create_ssd_models_from_config
[ RUN ] ModelBuilderTF1Test.test_invalid_faster_rcnn_batchnorm_update
[ OK ] ModelBuilderTF1Test.test_invalid_faster_rcnn_batchnorm_update
[ RUN ] ModelBuilderTF1Test.test_invalid_first_stage_nms_iou_threshold
[ OK ] ModelBuilderTF1Test.test_invalid_first_stage_nms_iou_threshold
[ RUN ] ModelBuilderTF1Test.test_invalid_model_config_proto
[ OK ] ModelBuilderTF1Test.test_invalid_model_config_proto
[ RUN ] ModelBuilderTF1Test.test_invalid_second_stage_batch_size
[ OK ] ModelBuilderTF1Test.test_invalid_second_stage_batch_size
[ RUN ] ModelBuilderTF1Test.test_session
[ SKIPPED ] ModelBuilderTF1Test.test_session
[ RUN ] ModelBuilderTF1Test.test_unknown_faster_rcnn_feature_extractor
[ OK ] ModelBuilderTF1Test.test_unknown_faster_rcnn_feature_extractor
[ RUN ] ModelBuilderTF1Test.test_unknown_meta_architecture
[ OK ] ModelBuilderTF1Test.test_unknown_meta_architecture
[ RUN ] ModelBuilderTF1Test.test_unknown_ssd_feature_extractor
[ OK ] ModelBuilderTF1Test.test_unknown_ssd_feature_extractor
----------------------------------------------------------------------
Ran 21 tests in 0.293s
OK (skipped=1)