Upload folder using huggingface_hub
Browse files- models/hparams.json +9 -5
- models/mask_sam_auto_prompt_ViT-B_dice.png +0 -0
- models/mask_sam_auto_prompt_ViT-B_mask_aux_loss.png +0 -0
- models/mask_sam_auto_prompt_ViT-B_mask_final_loss.png +0 -0
- models/mask_sam_auto_prompt_ViT-B_total_loss.png +0 -0
- models/mask_sam_model_best.pth +2 -2
- models/mask_sam_model_latest.pth +2 -2
- models/test_log.csv +2 -0
- models/train_log.csv +10 -1
models/hparams.json
CHANGED
|
@@ -1,21 +1,25 @@
|
|
| 1 |
{
|
| 2 |
-
"num_epochs":
|
| 3 |
"batch_size": 4,
|
| 4 |
"use_amp": false,
|
| 5 |
"use_wandb": false,
|
| 6 |
-
"use_boxes":
|
| 7 |
"use_masks": true,
|
| 8 |
-
"patience":
|
| 9 |
"task_name": "mask_sam_auto_prompt_ViT-B",
|
| 10 |
"model_save_path": "./models",
|
| 11 |
"save_dir": "./epoch_outputs",
|
| 12 |
-
"num_saved":
|
| 13 |
"good_ex": [
|
| 14 |
20,
|
| 15 |
267,
|
| 16 |
511
|
| 17 |
],
|
|
|
|
|
|
|
| 18 |
"use_mask_aux_loss": true,
|
| 19 |
"use_mask_final_loss": true,
|
| 20 |
-
"use_box_loss":
|
|
|
|
|
|
|
| 21 |
}
|
|
|
|
| 1 |
{
|
| 2 |
+
"num_epochs": 10,
|
| 3 |
"batch_size": 4,
|
| 4 |
"use_amp": false,
|
| 5 |
"use_wandb": false,
|
| 6 |
+
"use_boxes": false,
|
| 7 |
"use_masks": true,
|
| 8 |
+
"patience": 3,
|
| 9 |
"task_name": "mask_sam_auto_prompt_ViT-B",
|
| 10 |
"model_save_path": "./models",
|
| 11 |
"save_dir": "./epoch_outputs",
|
| 12 |
+
"num_saved": 30,
|
| 13 |
"good_ex": [
|
| 14 |
20,
|
| 15 |
267,
|
| 16 |
511
|
| 17 |
],
|
| 18 |
+
"image_size": 1024,
|
| 19 |
+
"hidden_dim": 64,
|
| 20 |
"use_mask_aux_loss": true,
|
| 21 |
"use_mask_final_loss": true,
|
| 22 |
+
"use_box_loss": false,
|
| 23 |
+
"lr": 0.0001,
|
| 24 |
+
"weight_decay": 1e-05
|
| 25 |
}
|
models/mask_sam_auto_prompt_ViT-B_dice.png
ADDED
|
models/mask_sam_auto_prompt_ViT-B_mask_aux_loss.png
ADDED
|
models/mask_sam_auto_prompt_ViT-B_mask_final_loss.png
ADDED
|
models/mask_sam_auto_prompt_ViT-B_total_loss.png
ADDED
|
models/mask_sam_model_best.pth
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d1c920b898eb353b9518d380943c1e2e188baed3b1c2a7ca0397dd6e2a054a6b
|
| 3 |
+
size 382995246
|
models/mask_sam_model_latest.pth
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4170173d8a218174ff75aed872083093a4ff6b25baaa317ef3909c3e641af4d9
|
| 3 |
+
size 382996242
|
models/test_log.csv
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
datetime,test_loss,test_loss_mask_aux,test_loss_mask_final,test_loss_box,test_dice
|
| 2 |
+
20250928-1346,0.8697482796602471,0.2722656068067218,0.5974826729574869,,0.3152020002462831
|
models/train_log.csv
CHANGED
|
@@ -1,2 +1,11 @@
|
|
| 1 |
datetime,epoch,train_loss,train_aux,train_final,train_box,val_loss,val_aux,val_final,val_box,val_dice
|
| 2 |
-
20250928-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
datetime,epoch,train_loss,train_aux,train_final,train_box,val_loss,val_aux,val_final,val_box,val_dice
|
| 2 |
+
20250928-0342,0,1.406314584362598,0.5247652019177381,0.8815493833927708,0.0,1.1587886292922316,0.3910911676242267,0.7676974592475116,0.0,0.1246067844982038
|
| 3 |
+
20250928-0448,1,1.1184222366821979,0.3632997257810361,0.7551225112898052,0.0,1.0722901225090027,0.33067727474694325,0.7416128504094739,0.0,0.21452825303369102
|
| 4 |
+
20250928-0552,2,1.058423319390712,0.3323757939381909,0.726047525803248,0.0,1.0236710872746966,0.3160229165693225,0.7076481701758912,0.0,0.2634861872594335
|
| 5 |
+
20250928-0657,3,1.0079862510278328,0.31024369151442865,0.6977425589588763,0.0,0.9896835654520141,0.29317791143498445,0.6965056532228053,0.0,0.20115270837546806
|
| 6 |
+
20250928-0803,4,0.9697688368288918,0.29414661113959534,0.6756222268552271,0.0,0.9779808923677744,0.30480322029989987,0.6731776747909294,0.0,0.2407389787669284
|
| 7 |
+
20250928-0909,5,0.9448810548879415,0.28690221319425346,0.6579788418595724,0.0,0.9528584794949759,0.29112842172232983,0.6617300602687797,0.0,0.3331213096046801
|
| 8 |
+
20250928-1014,6,0.9196709902535262,0.279996883934577,0.6396741062099394,0.0,0.9241589823652645,0.27873278999237844,0.6454261942260762,0.0,0.33571232587771566
|
| 9 |
+
20250928-1121,7,0.8967888621807705,0.27589820753830385,0.6208906547941324,0.0,0.9373658438624465,0.29442720171915093,0.6429386456605747,0.0,0.37393682372524656
|
| 10 |
+
20250928-1228,8,0.8758336915314653,0.2705109796839892,0.6053227119090903,0.0,0.8961917910781608,0.27190501874473494,0.6242867717283026,0.0,0.2736834118371049
|
| 11 |
+
20250928-1334,9,0.8507596660812999,0.26151800952832815,0.5892416565245344,0.0,0.8683019036871528,0.27399637606845895,0.5943055279968958,0.0,0.35238232163932215
|