抄写YOLOE源码——先抄写ultralytics包,关于__init__.py

什么是__init__.py文件呢,这个文件是一个标志。比如有一个文件夹的名字叫做"ultralytics",里面如果有了__init__.py文件,pip就会认为ultralytics文件夹是一个pip包,这个文件只能叫这个名字,因为pip只识别这个名字,换个名字就不行了。

再看这个文件里面,有一个__version__ = "8.3.39"。这一行表示版本号,版本号的变量名只能叫做__version__,因为pip只识别这个名字,换个名字就不行了。版本号的格式也只能是几点几点几,不能是别的。

然后是设置环境变量:

if not os.environ.get("OMP_NUM_THREADS"):

os.environ["OMP_NUM_THREADS"] = "1"

这个环境变量表示的是允许的线程数,从代码中可知,至少也应该有一个线程。

然后就是导包代码:

from ultralytics.models import NAS, RTDETR, SAM, YOLO, FastSAM, YOLOE

from ultralytics.utils import ASSETS, SETTINGS

from ultralytics.utils.checks import check_yolo as checks

from ultralytics.utils.downloads import download

再然后,settings = SETTINGS,这行代码是起别名

最后:

all = (

"version",

"ASSETS",

"YOLO",

"YOLOE",

"NAS",

"SAM",

"FastSAM",

"RTDETR",

"checks",

"download",

"settings",

)

这个表示从ultralytics这个包中都能import什么

相关推荐
Alan GEO实施教练2 小时前
专利申请是否找代理机构:核心考量与决策逻辑拆解
大数据·人工智能·python
格林威2 小时前
Baumer相机铝箔表面针孔检测:提升包装阻隔性的 7 个核心策略,附 OpenCV+Halcon 实战代码!
开发语言·人工智能·数码相机·opencv·计算机视觉·c#·工业相机
日出等日落2 小时前
用 Kavita实现我的远程数字书屋搭建记!
java·开发语言·ide·vscode·编辑器
berryyan2 小时前
🚀 Windows + RTX 5090 + ComfyUI 桌面版 安装 SageAttention 完全手册
人工智能·python
我是伪码农2 小时前
JS 复习
开发语言·前端·javascript
沐知全栈开发2 小时前
键盘快捷键:提升工作效率的利器
开发语言
性感程序员在线debug2 小时前
Python yield 原理
python
野犬寒鸦2 小时前
高并发利器:SingleFlight优化指南(Java版实现与项目实战)
服务器·开发语言·redis·后端·面试
Alan GEO实施教练2 小时前
实用新型专利申请代理机构选择:关键考量因素与实操要点讲解
java·开发语言·python