Browse Source

Created using Colaboratory

modifyDataloader
Glenn Jocher 3 years ago
parent
commit
b27f69f108
1 changed files with 31 additions and 7 deletions
  1. +31
    -7
      tutorial.ipynb

+ 31
- 7
tutorial.ipynb View File

@@ -19,6 +19,7 @@
"2e915d9016c846e095e382b6a02ee773": {
"model_module": "@jupyter-widgets/controls",
"model_name": "HBoxModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "HBoxView",
"_dom_classes": [],
@@ -39,6 +40,7 @@
"cb7fc3a5c6cc4fde8d2c83e594a7c86e": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
@@ -90,6 +92,7 @@
"ac3edef4e3434f4587e6cbf8aa048770": {
"model_module": "@jupyter-widgets/controls",
"model_name": "FloatProgressModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "ProgressView",
"style": "IPY_MODEL_13842ca90c0047e584b8d68d99dad2b1",
@@ -113,6 +116,7 @@
"853ac234cc2a4236946fc516871e10eb": {
"model_module": "@jupyter-widgets/controls",
"model_name": "HTMLModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "HTMLView",
"style": "IPY_MODEL_f94a7ca8c1f04761bf38fdc5f99664b8",
@@ -133,6 +137,7 @@
"13842ca90c0047e584b8d68d99dad2b1": {
"model_module": "@jupyter-widgets/controls",
"model_name": "ProgressStyleModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "StyleView",
"_model_name": "ProgressStyleModel",
@@ -148,6 +153,7 @@
"f454999c3a924c7bad0746fb453dec36": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
@@ -199,6 +205,7 @@
"f94a7ca8c1f04761bf38fdc5f99664b8": {
"model_module": "@jupyter-widgets/controls",
"model_name": "DescriptionStyleModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "StyleView",
"_model_name": "DescriptionStyleModel",
@@ -213,6 +220,7 @@
"9da1a23b042c41618dd14b0e30aa7cbe": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
@@ -264,6 +272,7 @@
"6ff8a710ded44391a624dec5c460b771": {
"model_module": "@jupyter-widgets/controls",
"model_name": "HBoxModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "HBoxView",
"_dom_classes": [],
@@ -284,6 +293,7 @@
"3c19729b51cd45d4848035da06e96ff8": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
@@ -335,6 +345,7 @@
"23b2f0ae3d46438c8de375987c77f580": {
"model_module": "@jupyter-widgets/controls",
"model_name": "FloatProgressModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "ProgressView",
"style": "IPY_MODEL_d8dda4b2ce864fd682e558b9a48f602e",
@@ -358,6 +369,7 @@
"dd9498c321a9422da6faf17a0be026d4": {
"model_module": "@jupyter-widgets/controls",
"model_name": "HTMLModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "HTMLView",
"style": "IPY_MODEL_0f84fe609bcf4aa9afdc32a8cf076909",
@@ -378,6 +390,7 @@
"d8dda4b2ce864fd682e558b9a48f602e": {
"model_module": "@jupyter-widgets/controls",
"model_name": "ProgressStyleModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "StyleView",
"_model_name": "ProgressStyleModel",
@@ -393,6 +406,7 @@
"ff8151449e444a14869684212b9ab14e": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
@@ -444,6 +458,7 @@
"0f84fe609bcf4aa9afdc32a8cf076909": {
"model_module": "@jupyter-widgets/controls",
"model_name": "DescriptionStyleModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "StyleView",
"_model_name": "DescriptionStyleModel",
@@ -458,6 +473,7 @@
"8fda673769984e2b928ef820d34c85c3": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
@@ -564,7 +580,7 @@
"clear_output()\n",
"print(f\"Setup complete. Using torch {torch.__version__} ({torch.cuda.get_device_properties(0).name if torch.cuda.is_available() else 'CPU'})\")"
],
"execution_count": 1,
"execution_count": null,
"outputs": [
{
"output_type": "stream",
@@ -585,7 +601,15 @@
"\n",
"`detect.py` runs YOLOv5 inference on a variety of sources, downloading models automatically from the [latest YOLOv5 release](https://github.com/ultralytics/yolov5/releases), and saving results to `runs/detect`. Example inference sources are:\n",
"\n",
"<img align=\"left\" src=\"https://user-images.githubusercontent.com/26833433/114307955-5c7e4e80-9ae2-11eb-9f50-a90e39bee53f.png\" width=\"900\"> "
"```shell\n",
"python detect.py --source 0 # webcam\n",
" file.jpg # image \n",
" file.mp4 # video\n",
" path/ # directory\n",
" path/*.jpg # glob\n",
" 'https://youtu.be/NUsoVlDFqZg' # YouTube\n",
" 'rtsp://example.com/media.mp4' # RTSP, RTMP, HTTP stream\n",
"```"
]
},
{
@@ -601,7 +625,7 @@
"!python detect.py --weights yolov5s.pt --img 640 --conf 0.25 --source data/images/\n",
"Image(filename='runs/detect/exp/zidane.jpg', width=600)"
],
"execution_count": 9,
"execution_count": null,
"outputs": [
{
"output_type": "stream",
@@ -675,7 +699,7 @@
"torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/releases/download/v1.0/coco2017val.zip', 'tmp.zip')\n",
"!unzip -q tmp.zip -d ../datasets && rm tmp.zip"
],
"execution_count": 10,
"execution_count": null,
"outputs": [
{
"output_type": "display_data",
@@ -715,7 +739,7 @@
"# Run YOLOv5x on COCO val2017\n",
"!python val.py --weights yolov5x.pt --data coco.yaml --img 640 --iou 0.65 --half"
],
"execution_count": 11,
"execution_count": null,
"outputs": [
{
"output_type": "stream",
@@ -839,7 +863,7 @@
"torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/releases/download/v1.0/coco128.zip', 'tmp.zip')\n",
"!unzip -q tmp.zip -d ../ && rm tmp.zip"
],
"execution_count": 12,
"execution_count": null,
"outputs": [
{
"output_type": "display_data",
@@ -917,7 +941,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": 13,
"execution_count": null,
"outputs": [
{
"output_type": "stream",

Loading…
Cancel
Save