Backbone từ mô hình tiền huấn luyện

Đến lúc xây dựng kiến trúc R-CNN! Bạn sẽ dùng backbone của mô hình tiền huấn luyện vgg16 để trích xuất đặc trưng. Hãy nhớ lưu lại kích thước đầu ra của backbone vì nó sẽ là kích thước đầu vào cho các khối tiếp theo: bộ phân loại (classifier) và bộ hồi quy hộp (box regressor).

torch, torchvision, torch.nn dưới tên nn đã được import. Mô hình đã được import là vgg16 với trọng số lưu trong VGG16_Weights.