tangyuanhao commited on
Update README.md
Browse files
README.md
CHANGED
|
@@ -12,17 +12,4 @@ cd AFM-Net
|
|
| 12 |
|
| 13 |
# Usage
|
| 14 |
|
| 15 |
-
|
| 16 |
-
from your_model_file import AdvancedFusionModel # Make sure to have your model definition available
|
| 17 |
-
|
| 18 |
-
|
| 19 |
-
|
| 20 |
-
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
model = (num_classes=45) # Example for NWPU-RESISC45, adjust as needed
|
| 24 |
-
|
| 25 |
-
model_path = "path/to/your/model_weights.pth" # e.g., 'uc_aid_nwpu_weights.pth'
|
| 26 |
-
model.load_state_dict(torch.load(model_path, map_location=device))
|
| 27 |
-
model.to(device)
|
| 28 |
-
model.eval()
|
|
|
|
| 12 |
|
| 13 |
# Usage
|
| 14 |
|
| 15 |
+
python app.py
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|