From a5b70a838f75c47e64a778c1548f5f213c09a1fa Mon Sep 17 00:00:00 2001 From: bionickatana Date: Sat, 11 Jul 2026 11:21:22 -0600 Subject: [PATCH] adding requirements.txt --- YOLO_model.ipynb | 239 ++++++++++++++++++++++++++++++++++++++++++++++- requirements.txt | 7 ++ 2 files changed, 244 insertions(+), 2 deletions(-) create mode 100644 requirements.txt diff --git a/YOLO_model.ipynb b/YOLO_model.ipynb index 1d36539..3975d5b 100644 --- a/YOLO_model.ipynb +++ b/YOLO_model.ipynb @@ -163,7 +163,242 @@ " all 200 688 0.359 0.237 0.224 0.108\n", "\n", " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", - "\u001b[K 8/75 0G 1.783 2.766 1.169 124 640: 76% ━━━━━━━━━─── 10/13 9.0s/it 1:36<27.1s" + "\u001b[K 8/75 0G 1.787 2.741 1.171 72 640: 100% ━━━━━━━━━━━━ 13/13 8.4s/it 1:496.6ss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 2.9s/it 20.0s3.6ss\n", + " all 200 688 0.362 0.219 0.277 0.0885\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 9/75 0G 1.756 2.733 1.159 77 640: 100% ━━━━━━━━━━━━ 13/13 7.6s/it 1:396.4ss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 2.9s/it 20.2s3.7ss\n", + " all 200 688 0.426 0.212 0.31 0.169\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 10/75 0G 1.684 2.538 1.139 52 640: 100% ━━━━━━━━━━━━ 13/13 7.6s/it 1:396.3ss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 2.9s/it 20.3s3.7ss\n", + " all 200 688 0.44 0.287 0.311 0.096\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 11/75 0G 1.673 2.409 1.128 40 640: 100% ━━━━━━━━━━━━ 13/13 7.7s/it 1:406.3ss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 2.8s/it 19.7s3.5ss\n", + " all 200 688 0.386 0.303 0.347 0.175\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 12/75 0G 1.71 2.465 1.146 65 640: 100% ━━━━━━━━━━━━ 13/13 7.6s/it 1:386.4ss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 2.8s/it 19.9s3.6ss\n", + " all 200 688 0.38 0.39 0.37 0.209\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 13/75 0G 1.633 2.348 1.12 61 640: 100% ━━━━━━━━━━━━ 13/13 7.5s/it 1:376.3ss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 2.9s/it 20.3s3.7ss\n", + " all 200 688 0.359 0.436 0.365 0.155\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 14/75 0G 1.57 2.291 1.113 55 640: 100% ━━━━━━━━━━━━ 13/13 5.3s/it 1:093.8ss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.8s/it 12.9s2.3s\n", + " all 200 688 0.428 0.487 0.425 0.249\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 15/75 0G 1.502 2.204 1.105 60 640: 100% ━━━━━━━━━━━━ 13/13 4.8s/it 1:023.8s0s\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.8s/it 12.7s2.3s\n", + " all 200 688 0.395 0.488 0.425 0.215\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 16/75 0G 1.665 2.212 1.134 52 640: 100% ━━━━━━━━━━━━ 13/13 4.8s/it 1:034.1sss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.8s/it 12.4s2.2s\n", + " all 200 688 0.51 0.374 0.456 0.28\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 17/75 0G 1.57 2.146 1.12 52 640: 100% ━━━━━━━━━━━━ 13/13 4.8s/it 1:023.9s4s\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 12.2s2.2s\n", + " all 200 688 0.495 0.447 0.458 0.28\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 18/75 0G 1.499 2.142 1.101 74 640: 100% ━━━━━━━━━━━━ 13/13 4.9s/it 1:033.9sss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 12.2s2.2s\n", + " all 200 688 0.423 0.456 0.455 0.286\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 19/75 0G 1.461 2.083 1.068 55 640: 100% ━━━━━━━━━━━━ 13/13 4.9s/it 1:043.9sss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 12.1s2.2s\n", + " all 200 688 0.477 0.499 0.486 0.305\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 20/75 0G 1.469 1.978 1.085 56 640: 100% ━━━━━━━━━━━━ 13/13 4.9s/it 1:044.0sss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 12.0s2.2s\n", + " all 200 688 0.47 0.603 0.511 0.312\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 21/75 0G 1.472 1.938 1.084 51 640: 100% ━━━━━━━━━━━━ 13/13 4.8s/it 1:023.8s2s\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 11.8s2.1s\n", + " all 200 688 0.483 0.579 0.518 0.32\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 22/75 0G 1.478 1.914 1.069 52 640: 100% ━━━━━━━━━━━━ 13/13 4.7s/it 1:013.8sss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 12.0s2.2s\n", + " all 200 688 0.453 0.522 0.476 0.206\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 23/75 0G 1.453 1.942 1.073 107 640: 100% ━━━━━━━━━━━━ 13/13 4.7s/it 1:013.8sss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 11.7s2.1s\n", + " all 200 688 0.506 0.599 0.534 0.355\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 24/75 0G 1.412 1.843 1.064 60 640: 100% ━━━━━━━━━━━━ 13/13 4.7s/it 1:013.8sss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 12.1s2.2s\n", + " all 200 688 0.466 0.601 0.559 0.311\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 25/75 0G 1.388 1.827 1.048 60 640: 100% ━━━━━━━━━━━━ 13/13 4.7s/it 1:023.8s1s\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 12.0s2.2s\n", + " all 200 688 0.501 0.627 0.592 0.377\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 26/75 0G 1.517 1.824 1.093 32 640: 100% ━━━━━━━━━━━━ 13/13 4.7s/it 1:013.8s1s\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 12.1s2.2s\n", + " all 200 688 0.544 0.606 0.587 0.352\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 27/75 0G 1.407 1.808 1.053 45 640: 100% ━━━━━━━━━━━━ 13/13 4.7s/it 1:023.8s1s\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 11.8s2.1s\n", + " all 200 688 0.566 0.621 0.615 0.415\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 28/75 0G 1.368 1.76 1.036 30 640: 100% ━━━━━━━━━━━━ 13/13 4.7s/it 1:023.8sss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 12.1s2.2s\n", + " all 200 688 0.495 0.678 0.625 0.413\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 29/75 0G 1.404 1.798 1.041 32 640: 100% ━━━━━━━━━━━━ 13/13 4.7s/it 1:023.9s2s\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 12.1s2.2s\n", + " all 200 688 0.548 0.61 0.601 0.306\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 30/75 0G 1.368 1.682 1.05 40 640: 100% ━━━━━━━━━━━━ 13/13 4.7s/it 1:013.8s1s\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 12.0s2.2s\n", + " all 200 688 0.661 0.586 0.607 0.382\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 31/75 0G 1.299 1.681 1.017 51 640: 100% ━━━━━━━━━━━━ 13/13 4.7s/it 1:023.8s1s\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 11.9s2.2s\n", + " all 200 688 0.544 0.656 0.625 0.41\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 32/75 0G 1.329 1.677 1.027 50 640: 100% ━━━━━━━━━━━━ 13/13 4.7s/it 1:013.8s0s\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 11.8s2.1s\n", + " all 200 688 0.574 0.685 0.673 0.454\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 33/75 0G 1.328 1.649 1.018 47 640: 100% ━━━━━━━━━━━━ 13/13 4.7s/it 1:013.8s0s\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 12.1s2.2s\n", + " all 200 688 0.625 0.666 0.681 0.433\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 34/75 0G 1.354 1.647 1.01 76 640: 100% ━━━━━━━━━━━━ 13/13 4.7s/it 1:013.8s1s\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 12.0s2.2s\n", + " all 200 688 0.578 0.642 0.646 0.355\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 35/75 0G 1.354 1.638 1.044 60 640: 100% ━━━━━━━━━━━━ 13/13 4.7s/it 1:013.8s1s\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 11.9s2.1s\n", + " all 200 688 0.562 0.693 0.662 0.448\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 36/75 0G 1.438 1.563 1.054 70 640: 100% ━━━━━━━━━━━━ 13/13 4.8s/it 1:023.8s1s\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 11.7s2.1s\n", + " all 200 688 0.607 0.634 0.674 0.391\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 37/75 0G 1.382 1.563 1.03 55 640: 100% ━━━━━━━━━━━━ 13/13 4.7s/it 1:023.8s0s\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 12.2s2.2s\n", + " all 200 688 0.623 0.643 0.682 0.437\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 38/75 0G 1.451 1.534 1.038 71 640: 100% ━━━━━━━━━━━━ 13/13 4.7s/it 1:023.8s1s\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 12.1s2.2s\n", + " all 200 688 0.574 0.679 0.694 0.482\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 39/75 0G 1.306 1.489 1.012 61 640: 100% ━━━━━━━━━━━━ 13/13 4.7s/it 1:013.8s1s\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 12.0s2.2s\n", + " all 200 688 0.614 0.608 0.668 0.296\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 40/75 0G 1.314 1.573 1.026 30 640: 100% ━━━━━━━━━━━━ 13/13 4.7s/it 1:023.8s1s\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 11.7s2.1s\n", + " all 200 688 0.632 0.714 0.741 0.514\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 41/75 0G 1.347 1.474 1.031 66 640: 100% ━━━━━━━━━━━━ 13/13 4.7s/it 1:013.8s0s\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 11.8s2.1s\n", + " all 200 688 0.649 0.712 0.74 0.462\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 42/75 0G 1.497 1.432 1.085 50 640: 100% ━━━━━━━━━━━━ 13/13 4.7s/it 1:013.8sss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 12.2s2.2s\n", + " all 200 688 0.647 0.689 0.721 0.315\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 43/75 0G 1.39 1.428 1.053 39 640: 100% ━━━━━━━━━━━━ 13/13 4.7s/it 1:013.8s1s\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.7s/it 12.0s2.2s\n", + " all 200 688 0.665 0.724 0.757 0.539\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 44/75 0G 1.383 1.464 1.043 69 640: 100% ━━━━━━━━━━━━ 13/13 6.1s/it 1:195.8ss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 2.6s/it 18.0s3.2ss\n", + " all 200 688 0.663 0.714 0.74 0.49\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 45/75 0G 1.314 1.475 1.007 21 640: 100% ━━━━━━━━━━━━ 13/13 6.9s/it 1:295.4ss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 2.4s/it 17.1s3.1ss\n", + " all 200 688 0.653 0.699 0.736 0.374\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 46/75 0G 1.277 1.467 1.014 39 640: 100% ━━━━━━━━━━━━ 13/13 5.8s/it 1:154.5ss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.9s/it 13.1s2.4s\n", + " all 200 688 0.662 0.717 0.755 0.525\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 47/75 0G 1.274 1.427 1.012 37 640: 100% ━━━━━━━━━━━━ 13/13 5.1s/it 1:064.3sss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.9s/it 13.6s2.4s\n", + " all 200 688 0.702 0.713 0.776 0.539\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 48/75 0G 1.199 1.375 0.9964 29 640: 100% ━━━━━━━━━━━━ 13/13 5.0s/it 1:064.1sss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.8s/it 12.3s2.2s\n", + " all 200 688 0.657 0.737 0.769 0.484\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 49/75 0G 1.246 1.444 0.9904 47 640: 100% ━━━━━━━━━━━━ 13/13 6.3s/it 1:224.3ss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.8s/it 12.6s2.3s\n", + " all 200 688 0.668 0.753 0.782 0.523\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 50/75 0G 1.255 1.398 0.9953 40 640: 100% ━━━━━━━━━━━━ 13/13 5.9s/it 1:164.4ss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 2.0s/it 13.9s2.5s\n", + " all 200 688 0.651 0.742 0.78 0.514\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 51/75 0G 1.217 1.396 1.002 28 640: 100% ━━━━━━━━━━━━ 13/13 5.9s/it 1:174.6ss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.8s/it 12.7s2.3s\n", + " all 200 688 0.688 0.713 0.768 0.496\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 52/75 0G 1.148 1.324 0.9712 49 640: 100% ━━━━━━━━━━━━ 13/13 5.1s/it 1:064.3sss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.9s/it 13.2s2.5s\n", + " all 200 688 0.699 0.727 0.793 0.559\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 53/75 0G 1.197 1.306 0.9847 44 640: 100% ━━━━━━━━━━━━ 13/13 5.2s/it 1:084.1sss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 1.8s/it 12.8s2.3s\n", + " all 200 688 0.679 0.774 0.794 0.546\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 54/75 0G 1.164 1.337 0.9704 39 640: 100% ━━━━━━━━━━━━ 13/13 5.2s/it 1:074.9sss\n", + "\u001b[K Class Images Instances Box(P R mAP50 mAP50-95): 100% ━━━━━━━━━━━━ 7/7 2.6s/it 18.2s3.3ss\n", + " all 200 688 0.674 0.739 0.768 0.55\n", + "\n", + " Epoch GPU_mem box_loss cls_loss dfl_loss Instances Size\n", + "\u001b[K 55/75 0G 1.165 1.293 0.9786 126 640: 23% ━━╸───────── 3/13 11.5s/it 29.9s<1:55" ] } ], @@ -185,7 +420,7 @@ "metadata": {}, "outputs": [], "source": [ - "model = YOLO(\"runs/detect/train/weights/best.pt\")" + "# model = YOLO(\"runs/detect/train/weights/best.pt\")" ] }, { diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..f00e120 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,7 @@ +opencv-python +tensorflow +pillow +scipy +matplotlib +jupyter +ultralytics