qsbye/runtime-env7 repository overview

#arm64版本
docker run -it -v $PWD:/srv -p 6006:6006 qsbye/runtime-env7:v0.2-arm64 bash
cd /home/pytorch-ssd
#可以运行,但是非常慢
python3 train_ssd.py --datasets /srv/files3/VOC2007 --validation_dataset /srv/files3/VOC2007_test --net mb1-ssd --base_net ./models/mobilenet_v1_with_relu_69_5.pth  --batch_size 12 --num_epochs 100 --scheduler cosine --lr 0.01 --t_max 200 --num_workers 1 --use_cuda False

#基于yolov5-arm64构建
docker run -it --network=bridge -v $PWD:/srv -p 6006:6006 qsbye/runtime-env5:v0.1-arm64 bash
pip list
python3 -m pip install --upgrade pip
apt update
pip install caffe2
cp -r /srv/files3/pytorch-ssd /home
cp /srv/files3/pytorch-ssd/cat.jpg /home/pytorch-ssd
ls /home
cd /home/pytorch-ssd
cp -r /srv/files3/coco2voc /home
pip install pycocotools
cd /home/coco2voc
python3 coco2voc.py --ann_file /srv/files3/files3/labelstudio-data/media/upload/display/coco.json --output_dir /srv/files3/files3/labelstudio-data/media/upload