Sign inSign up

robin990/keras-yolo3

By robin990

•Updated over 6 years ago

keras-yolo3 tensorflow1.7

Image
0

145

robin990/keras-yolo3 repository overview

监控: nvidia-smi

创建镜像: docker build -t keras-yolo3_tf:1.0 -f keras-yolo3_Dockerfile .

运行容器CPU:

bridge集群独立

docker run --network bridge -p 6006:6006 -p 23:22 -v /home/zy/workspace/docker_workspace_01/keras-yolo3-master:/home/zy/workspace/docker_workspace_01/keras-yolo3-master -dit --name ky0.3 keras-yolo3:0.3

host可以连外网

docker run --network host -v /home/zy/workspace/docker_workspace_01/keras-yolo3-master:/home/zy/workspace/docker_workspace_01/keras-yolo3-master -dit --name ky0.3 keras-yolo3:0.3

交互容器

docker exec -it ky0.3 /bin/bash

运行容器GPU:

bridge集群独立

nvidia-docker run --runtime=nvidia --network bridge -p 6007:6006 -p 24:22 -v /home/zy/workspace/docker_workspace_01/keras-yolo3-master:/home/zy/workspace/docker_workspace_01/keras-yolo3-master -dit --name ky0.3-gpu keras-yolo3:0.3-gpu

host可以连外网

nvidia-docker run --runtime=nvidia --network host -v /home/zy/workspace/docker_workspace_01/keras-yolo3-master:/home/zy/workspace/docker_workspace_01/keras-yolo3-master -dit --name ky0.3-gpu keras-yolo3:0.3-gpu

交互容器

nvidia-docker exec -it ky0.3-gpu /bin/bash

训练: python train2.py

测试: python yolo_video_opencv.py --model_path /home/zy/workspace/docker_workspace_01/keras-yolo3-master/logs/PIL_111/ep260-loss68.298-val_loss83.504.h5 --input /home/zy/workspace/docker_workspace_01/keras-yolo3-master/VOCdevkit/VOC2007/JPEGImages/1_original_img_0.jpg --image

scp -P 23 [email protected]:/home/zy/.bashrc /home/zy/

Tag summary

Content type

Image

Digest

Size

1.6 GB

Last updated

over 6 years ago

docker pull robin990/keras-yolo3:0.3-gpu