Update README.md
Browse files
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 |
+
|