anomani13 commited on
Commit
27672e7
·
verified ·
1 Parent(s): 56daab4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +24 -0
README.md CHANGED
@@ -5,3 +5,27 @@ lon_std=0.0004335530163099028
5
  ---
6
  license: mit
7
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
  ---
6
  license: mit
7
  ---
8
+
9
+
10
+ #ResNet GPS Prediction Model
11
+ This model predicts GPS coordinates (latitude and longitude) from input images using a custom ResNet-based architecture.
12
+
13
+ ##How to Use
14
+ 1. Download `resnet_gps_model.pth` and `config.json` from this repository.
15
+ 2. Define the model architecture (as shown in the usage example below).
16
+ 3. Load the model weights and configuration.
17
+
18
+ ## Example Usage
19
+ ```python
20
+ import torch
21
+ import json
22
+
23
+ # Load config
24
+ config = json.load(open("config.json", "r"))
25
+
26
+ # Define and load model
27
+ resnet = CustomResNetModel(model_name="microsoft/resnet-18", num_classes=config["num_classes"])
28
+ state_dict = torch.load("resnet_gps_model.pth")
29
+ resnet.load_state_dict(state_dict)
30
+ resnet.eval()
31
+