yolov5_sagemaker

1. Build and Push

./build_and_push yolov5

./train_local.sh

./predicct_local.sh

Refer to yolov5_byoc.ipynb, please note that SageMaker local test mode is not supported, because of the shared memory reason.

Name		Name	Last commit message	Last commit date
Latest commit History 31 Commits
data		data
lambda		lambda
model-classifier/code		model-classifier/code
model/code		model/code
utils		utils
.gitignore		.gitignore
Dockerfile		Dockerfile
README.md		README.md
build_and_push.sh		build_and_push.sh
create_endpoint.py		create_endpoint.py
local_predict-x-image.py		local_predict-x-image.py
local_predict.py		local_predict.py
nginx.conf		nginx.conf
predict		predict
predict_local.sh		predict_local.sh
predictor.py		predictor.py
serve		serve
test_api.sh		test_api.sh
train		train
train_local.sh		train_local.sh
wsgi.py		wsgi.py
yolov5_byoc.ipynb		yolov5_byoc.ipynb
yolov5_byos.ipynb		yolov5_byos.ipynb
yolov5_classifier_byos.ipynb		yolov5_classifier_byos.ipynb