Khối hồi quy hộp (Box regressor)

Nhiệm vụ cuối cùng của bạn là tạo một khối hồi quy để dự đoán các tọa độ của bounding box. Bạn quyết định dùng một khối gồm 2 lớp fully connected với một kích hoạt ReLU ở giữa, tương tự bộ phân loại bạn đã định nghĩa trước đó.

vgg_model và input_dim của bạn vẫn có sẵn, và torch cùng torchvision.models đã được nhập.