CSPNet (CSPResNet, CSPDarkNet, DarkNet)

Thể loại: CNN · Code: timm/models/cspnet.py · Kiến trúc: 23 · Trọng số tiền huấn luyện: 15 (15 có kết quả trên ImageNet-1k)

Cross Stage Partial Networks chia các đặc trưng của mỗi giai đoạn thành hai phần, một phần đi qua các khối của giai đoạn và một phần được gộp lại ở cuối giai đoạn. Module này chứa CSPResNet, CSPResNeXt, CSPDarkNet, DarkNet và các biến thể cs3* của timm.

Bài báo

• CSPNet: Một Backbone mới có thể tăng cường khả năng học của CNN
• YOLOv3: Một cải tiến tăng dần

Cách sử dụng

Xem Quickstart để tải trọng số được huấn luyện trước, tiền xử lý ảnh và tinh chỉnh, và Feature Extraction để sử dụng các mô hình làm bộ xương sống.

>>> import timm
>>> model = timm.create_model('cspdarknet53.ra_in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)

Hỗ trợ API mô hình

✓ được hỗ trợ, ✓ (vuông) chỉ hoạt động với kích thước vuông, ✗ không được hỗ trợ, – không cần thiết (mô hình đã chấp nhận bất kỳ kích thước đầu vào nào). Được kiểm tra bằng cách chạy từng tính năng trên một mô hình được khởi tạo ngẫu nhiên (CspNet với cs3darknet_focus_s ở 256), các kiến trúc khác cùng lớp được kỳ vọng sẽ hoạt động tương tự. Xem trang Model API để biết chi tiết.

Kiến trúc

GMACs và số lượng tham số được tính cho kích thước đầu vào và bộ phân loại mặc định của mỗi kiến trúc, từ kết quả benchmark.

Trọng số được huấn luyện trước

Top-1 là độ chính xác kiểm tra trên ImageNet-1k từ results-imagenet.csv ở kích thước eval được liệt kê, - nếu trọng số không có kết quả ImageNet-1k (ví dụ trọng số pretraining hoặc image-text). Classes là kích thước của bộ phân loại được huấn luyện trước, không có đối với các trọng số không có bộ phân loại.

Trọng số được nhóm theo nguồn: bài báo hoặc nguồn gốc được nêu trong cấu hình tiền huấn luyện, nếu không thì một tổ chức Hub khác với timm, nếu không thì phần đầu của thẻ tiền huấn luyện (architecture.tag). Các nguồn chỉ có một trọng số được liệt kê dưới other.

c2ns tags

ra thẻ

Khác