Hugging Face's logo

nakamura196
/
yolov8-ndl-layout

Object Detection
PyTorch
ultralyticsplus
yolov8
yolov8x
yolo
vision
Model card Files Community
yolov8-ndl-layout
File size: 2,189 Bytes 
124cdc8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
 
{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## How tu use\n",
    "\n",
    "- Install [ultralyticsplus](https://github.com/fcakyon/ultralyticsplus):\n",
    "\n",
    "```bash\n",
    "pip install -U ultralyticsplus\n",
    "```\n",
    "\n",
    "- Load model and perform prediction:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\n",
      "Found https://dl.ndl.go.jp/api/iiif/2534020/T0000001/full/full/0/default.jpg locally at default.jpg\n",
      "image 1/1 /Users/nakamura/git/hf/models/yolov8-ndl-layout/default.jpg: 640x640 1 1_overall, 1 3_typography, 510.9ms\n",
      "Speed: 2.2ms preprocess, 510.9ms inference, 1.8ms postprocess per image at shape (1, 3, 640, 640)\n"
     ]
    }
   ],
   "source": [
    "from ultralyticsplus import YOLO, render_result\n",
    "import os\n",
    "\n",
    "# load model\n",
    "model = YOLO('model_- 19 may 2024 15_13.pt')\n",
    "  \n",
    "# set model parameters\n",
    "conf_threshold = 0.25  # NMS confidence threshold\n",
    "iou_threshold = 0.45  # NMS IoU threshold\n",
    "\n",
    "# set image\n",
    "img = 'https://dl.ndl.go.jp/api/iiif/2534020/T0000001/full/full/0/default.jpg'\n",
    "\n",
    "# perform inference\n",
    "results = model.predict(img, conf=conf_threshold, iou=iou_threshold)\n",
    "render = render_result(model=model, image=img, result=results[0])  \n",
    "\n",
    "os.makedirs('results', exist_ok=True)\n",
    "\n",
    "# save\n",
    "render.save('results/1.jpg')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": ".venv",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.9.11"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}