timm documentation

VoVNet

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v1.0.30).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

VoVNet

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

VoVNet sử dụng các module One-Shot Aggregation (OSA), trong đó các đầu ra của tất cả các lớp trong một block được nối (concatenate) lại với nhau một lần duy nhất ở cuối block đó. VoVNetV2 (từ CenterMask) bổ sung các kết nối dư (residual connections) và cơ chế chú ý SE hiệu quả (eSE attention).

Bài báo

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('ese_vovnet39b.ra_in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)

Hỗ trợ API mô hình

Tính năngVovNet
bản đồ đặc trưng features_only=True✓ (các bước nhảy 2, 4, 8, 16, 32)
forward_intermediates()✓
prune_intermediate_layers()✓
Bất kỳ kích thước đầu vào nào, không cần thay đổi✓
set_input_size()–
dynamic_img_size=True–
set_grad_checkpointing()✓
Suy giảm LR theo lớp (group_matcher())✓
reset_classifier()✓

✓ đượ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 (VovNet với ese_vovnet19b_slim_dw ở 224), 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

Kiến trúcSố tham số (M)GMACsKích thước đầu vàoTrọng số được huấn luyện trước
eca_vovnet39b22.607.092240
ese_vovnet19b_dw6.541.342241
ese_vovnet19b_slim3.171.692240
ese_vovnet19b_slim_dw1.900.402240
ese_vovnet39b24.577.092241
ese_vovnet39b_evos24.587.072240
ese_vovnet57b38.6111.692561
ese_vovnet99b63.2016.512240
vovnet39a22.607.092240
vovnet57a36.648.952240

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ốKích thước đầu vàoTop-1 (%)Kích thước đánh giáCác lớpGiấy phépHub
ese_vovnet19b_dw.ra_in1k224 (test 288)77.7882881000apache-2.0timm
ese_vovnet39b.ra_in1k224 (test 288)80.3782881000apache-2.0timm
ese_vovnet57b.ra4_e3600_r256_in1k256 (kiểm tra 320)83.2983201000apache-2.0timm
Update on GitHub