Tài liệu tham khảo cho ultralytics/hub/__init__.py
Ghi
Tệp này có sẵn tại https://github.com/ultralytics/ultralytics/blob/main/ultralytics/hub/init.py. Nếu bạn phát hiện ra một vấn đề, vui lòng giúp khắc phục nó bằng cách đóng góp Yêu cầu 🛠️ kéo. Cảm ơn bạn 🙏 !
ultralytics.hub.login(api_key=None, save=True)
Đăng nhập vào Ultralytics API HUB sử dụng khóa API được cung cấp.
Phiên không được lưu trữ; một phiên mới được tạo khi cần bằng cách sử dụng SETTINGS đã lưu hoặc HUB_API_KEY biến môi trường nếu được xác thực thành công.
Thông số:
Tên | Kiểu | Sự miêu tả | Mặc định |
---|---|---|---|
api_key |
str
|
Khóa API để sử dụng để xác thực. Nếu không được cung cấp, nó sẽ được truy xuất từ SETTINGS hoặc biến môi trường HUB_API_KEY. |
None
|
save |
bool
|
Có nên lưu khóa API vào CÀI ĐẶT nếu xác thực thành công hay không. |
True
|
Trở lại:
Kiểu | Sự miêu tả |
---|---|
bool
|
True nếu xác thực thành công, False nếu không. |
Mã nguồn trong ultralytics/hub/__init__.py
ultralytics.hub.logout()
Đăng xuất khỏi Ultralytics HUB bằng cách xóa khóa API khỏi tệp cài đặt. Để đăng nhập lại, hãy sử dụng 'yolo đăng nhập hub'.
Mã nguồn trong ultralytics/hub/__init__.py
ultralytics.hub.reset_model(model_id='')
Đặt lại mô hình đã đào tạo về trạng thái chưa được đào tạo.
Mã nguồn trong ultralytics/hub/__init__.py
ultralytics.hub.export_fmts_hub()
Trả về danh sách các định dạng xuất được HUB hỗ trợ.
ultralytics.hub.export_model(model_id='', format='torchscript')
Xuất mô hình sang tất cả các định dạng.
Mã nguồn trong ultralytics/hub/__init__.py
ultralytics.hub.get_export(model_id='', format='torchscript')
Nhận từ điển mô hình đã xuất với URL tải xuống.
Mã nguồn trong ultralytics/hub/__init__.py
ultralytics.hub.check_dataset(path, task)
Chức năng kiểm tra lỗi tệp Zip tập dữ liệu HUB trước khi tải lên. Nó kiểm tra một tập dữ liệu cho các lỗi trước khi nó được tải lên vào HUB. Ví dụ sử dụng được đưa ra dưới đây.
Thông số:
Tên | Kiểu | Sự miêu tả | Mặc định |
---|---|---|---|
path |
str
|
Đường dẫn đến data.zip (với data.yaml bên trong data.zip). |
bắt buộc |
task |
str
|
Tác vụ tập dữ liệu. Các tùy chọn là 'phát hiện', 'phân đoạn', 'tạo dáng', 'phân loại', 'obb'. |
bắt buộc |
Ví dụ
Tải xuống *.zip tệp từ https://github.com/ultralytics/hub/tree/main/example_datasets tức là https://github.com/ultralytics/hub/raw/main/example_datasets/coco8.zip cho coco8.zip.
from ultralytics.hub import check_dataset
check_dataset('path/to/coco8.zip', task='detect') # detect dataset
check_dataset('path/to/coco8-seg.zip', task='segment') # segment dataset
check_dataset('path/to/coco8-pose.zip', task='pose') # pose dataset
check_dataset('path/to/dota8.zip', task='obb') # OBB dataset
check_dataset('path/to/imagenet10.zip', task='classify') # classification dataset