Created using Colaboratory
parent
d631a6904e
commit
73ed7a5e7a
|
@ -415,7 +415,7 @@
|
|||
"import utils\n",
|
||||
"display = utils.notebook_init() # checks"
|
||||
],
|
||||
"execution_count": 1,
|
||||
"execution_count": null,
|
||||
"outputs": [
|
||||
{
|
||||
"output_type": "stream",
|
||||
|
@ -467,7 +467,7 @@
|
|||
"!python detect.py --weights yolov5s.pt --img 640 --conf 0.25 --source data/images\n",
|
||||
"#display.Image(filename='runs/detect/exp/zidane.jpg', width=600)"
|
||||
],
|
||||
"execution_count": 2,
|
||||
"execution_count": null,
|
||||
"outputs": [
|
||||
{
|
||||
"output_type": "stream",
|
||||
|
@ -547,7 +547,7 @@
|
|||
"torch.hub.download_url_to_file('https://ultralytics.com/assets/coco2017val.zip', 'tmp.zip')\n",
|
||||
"!unzip -q tmp.zip -d ../datasets && rm tmp.zip"
|
||||
],
|
||||
"execution_count": 3,
|
||||
"execution_count": null,
|
||||
"outputs": [
|
||||
{
|
||||
"output_type": "display_data",
|
||||
|
@ -578,7 +578,7 @@
|
|||
"# Run YOLOv5x on COCO val\n",
|
||||
"!python val.py --weights yolov5x.pt --data coco.yaml --img 640 --iou 0.65 --half"
|
||||
],
|
||||
"execution_count": 4,
|
||||
"execution_count": null,
|
||||
"outputs": [
|
||||
{
|
||||
"output_type": "stream",
|
||||
|
@ -753,7 +753,7 @@
|
|||
"# Train YOLOv5s on COCO128 for 3 epochs\n",
|
||||
"!python train.py --img 640 --batch 16 --epochs 3 --data coco128.yaml --weights yolov5s.pt --cache"
|
||||
],
|
||||
"execution_count": 5,
|
||||
"execution_count": null,
|
||||
"outputs": [
|
||||
{
|
||||
"output_type": "stream",
|
||||
|
@ -1131,7 +1131,6 @@
|
|||
},
|
||||
"source": [
|
||||
"# TensorRT \n",
|
||||
"# https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html#installing-pip\n",
|
||||
"!pip install -U nvidia-tensorrt --index-url https://pypi.ngc.nvidia.com # install\n",
|
||||
"!python export.py --weights yolov5s.pt --include engine --imgsz 640 --device 0 # export\n",
|
||||
"!python detect.py --weights yolov5s.engine --imgsz 640 --device 0 # inference"
|
||||
|
|
Loading…
Reference in New Issue