Phát hiện TensorFlow , Google Nền tảng ML mã nguồn mở mạnh mẽ dành cho đổi mới AI. Xây dựng, đào tạo và triển khai các mô hình mạng nơ-ron một cách liền mạch!
TensorFlow là một nền tảng mã nguồn mở toàn diện và linh hoạt được thiết kế để hợp lý hóa việc phát triển và triển khai các ứng dụng học máy (ML) và trí tuệ nhân tạo. Ban đầu được phát triển bởi các nhà nghiên cứu và kỹ sư từ nhóm Google Brain , nền tảng này đã phát triển thành một hệ sinh thái phong phú gồm các công cụ, thư viện và tài nguyên cộng đồng, cho phép các nhà nghiên cứu thúc đẩy công nghệ học sâu (DL) tiên tiến nhất, đồng thời cho phép các nhà phát triển dễ dàng xây dựng và triển khai các ứng dụng hỗ trợ ML. Kiến trúc của nền tảng được thiết kế linh hoạt, hỗ trợ tính toán trên nhiều nền tảng khác nhau, từ máy chủ mạnh mẽ đến thiết bị di động biên .
Ở trung tâm của nó, TensorFlow được xây dựng dựa trên khái niệm đồ thị luồng dữ liệu. Trong mô hình này, các nút trong đồ thị biểu diễn các phép toán, trong khi các cạnh của đồ thị biểu diễn các mảng dữ liệu đa chiều, được gọi là tenxơ, chảy giữa chúng. Kiến trúc này cho phép khung thực hiện các phép tính mạng nơ-ron (NN) phức tạp một cách hiệu quả.
Điểm mạnh của khuôn khổ này nằm ở hệ sinh thái mở rộng, hỗ trợ toàn bộ vòng đời ML từ xử lý dữ liệu trước cho đến triển khai sản xuất.
Trong bối cảnh của các khuôn khổ học sâu, sự so sánh chính thường được rút ra giữa TensorFlow và PyTorch . Mặc dù cả hai đều có khả năng xử lý khối lượng công việc nghiên cứu và sản xuất hiện đại, nhưng chúng có những khác biệt về mặt lịch sử. TensorFlow thường được ưa chuộng trong các thiết lập công nghiệp vì các đường ống triển khai mô hình mạnh mẽ và hỗ trợ phần cứng đa dạng thông qua các định dạng như SavedModel Và TFLite . PyTorch , do Meta phát triển, thường được trích dẫn nhờ đồ thị tính toán động và tính dễ sử dụng trong nghiên cứu học thuật. Tuy nhiên, với các bản cập nhật gần đây, khoảng cách đã thu hẹp đáng kể, và cả hai nền tảng đều mang lại khả năng tương tác và hiệu suất tuyệt vời.
Tính linh hoạt của khung này khiến nó phù hợp với nhiều ngành công nghiệp và nhiệm vụ phức tạp trong thị giác máy tính (CV) và xử lý ngôn ngữ tự nhiên.
Ultralytics YOLO các mô hình tích hợp liền mạch với TensorFlow hệ sinh thái. Người dùng có thể đào tạo các mô hình tiên tiến như YOLO11 trong Python và dễ dàng xuất chúng sang các định dạng tương thích để triển khai trên nền tảng web, di động hoặc đám mây. Khả năng này đảm bảo hiệu suất cao của YOLO có thể được tận dụng trong hiện tại TensorFlow -cơ sở hạ tầng dựa trên.
Ví dụ sau đây minh họa cách xuất một tệp đã được đào tạo trước YOLO11 mô hình theo định dạng TensorFlow SavedModel , cho phép tích hợp dễ dàng với các công cụ phục vụ.
from ultralytics import YOLO
# Load the official YOLO11 model
model = YOLO("yolo11n.pt")
# Export the model to TensorFlow SavedModel format
# This creates a directory containing the saved_model.pb file
model.export(format="saved_model")
Ngoài ra SavedModel , Ultralytics hỗ trợ xuất sang TensorFlow Lite cho các ứng dụng di động, TensorFlow .js để suy luận dựa trên web và Edge TPU để tăng tốc hiệu suất phần cứng.