Upload копия_блокнота__turtorial_ipynb_ (3).py

копия_блокнота__turtorial_ipynb_ (3).py

@@ -0,0 +1,35 @@
+# -*- coding: utf-8 -*-
+"""Копия_блокнота_"turtorial_ipynb".ipynb
+
+Automatically generated by Colaboratory.
+
+Original file is located at
+    https://colab.research.google.com/github/kisa-misa/YOLOv5_Torchvision-Yekaterinburg-crossroads-results/blob/main/%D0%9A%D0%BE%D0%BF%D0%B8%D1%8F_%D0%B1%D0%BB%D0%BE%D0%BA%D0%BD%D0%BE%D1%82%D0%B0_%22turtorial_ipynb%22.ipynb
+
+# Setup
+Clone repo and install dependencies.
+"""
+
+# Commented out IPython magic to ensure Python compatibility.
+!git clone https://github.com/meituan/YOLOv6.git
+# %cd YOLOv6
+# %pip install -r requirements.txt
+
+"""# Inference
+First, download a pretrained model from the YOLOv6 [release](https://github.com/meituan/YOLOv6/releases).
+
+### You can also use torch.hub style to load the pretrained model or custom model to inference.
+"""
+
+# Download a pretrained model
+import torch
+torch.hub.download_url_to_file('https://github.com/meituan/YOLOv6/releases/download/0.3.0/yolov6s.pt', 'yolov6s.pt')
+
+import torch 
+
+model_local = torch.hub.load('.', 'yolov6n', source='local')
+
+img_path = 'data/images/image1.jpg'
+prediction = model_local.predict(img_path)
+
+model_local.show_predict(img_path)