|
Dataset stats: \ |
|
lat_mean = 39.951564548022596 \ |
|
lat_std = 0.0006361722351128644 \ |
|
lon_mean = -75.19150880602636 \ |
|
lon_std = 0.000611411894337979 |
|
|
|
The model can be loaded using: |
|
``` |
|
from huggingface_hub import hf_hub_download |
|
import torch |
|
|
|
# Specify the repository and the filename of the model you want to load |
|
repo_id = "FinalProj5190/ImageToGPSproject-vit-base" # Replace with your repo name |
|
filename = "resnet_gps_regressor_complete.pth" |
|
|
|
model_path = hf_hub_download(repo_id=repo_id, filename=filename) |
|
|
|
# Load the model using torch |
|
model_test = torch.load(model_path) |
|
model_test.eval() # Set the model to evaluation mode |
|
``` |