Closed jimgoo closed 1 year ago
Add a class for embedding images with ResNet50, which takes a parameter for which layer to extract features from (either 2, 3, or 4). The MinReader paper uses layer 2 (with dim 512) as f_mr, then later in another experiment layer 4 (with dim 2048).
Add a class for embedding images with ResNet50, which takes a parameter for which layer to extract features from (either 2, 3, or 4). The MinReader paper uses layer 2 (with dim 512) as f_mr, then later in another experiment layer 4 (with dim 2048).