rawalkhirodkar commited on
Commit
2d4ff18
β€’
1 Parent(s): cc0c583

Add better checkpoints

Browse files
app.py CHANGED
@@ -71,9 +71,9 @@ class Config:
71
  ASSETS_DIR = os.path.join(os.path.dirname(__file__), 'assets')
72
  CHECKPOINTS_DIR = os.path.join(ASSETS_DIR, "checkpoints")
73
  CHECKPOINTS = {
74
- "0.3b": "sapiens_0.3b_goliath_best_goliath_AP_575_torchscript.pt2",
75
- "0.6b": "sapiens_0.6b_goliath_best_goliath_AP_600_torchscript.pt2",
76
- "1b": "sapiens_1b_goliath_best_goliath_AP_640_torchscript.pt2",
77
  }
78
  DETECTION_CHECKPOINT = os.path.join(CHECKPOINTS_DIR, 'rtmdet_m_8xb32-100e_coco-obj365-person-235e8209.pth')
79
  DETECTION_CONFIG = os.path.join(ASSETS_DIR, 'rtmdet_m_640-8xb32_coco-person_no_nms.py')
 
71
  ASSETS_DIR = os.path.join(os.path.dirname(__file__), 'assets')
72
  CHECKPOINTS_DIR = os.path.join(ASSETS_DIR, "checkpoints")
73
  CHECKPOINTS = {
74
+ "0.3b": "sapiens_0.3b_goliath_best_goliath_AP_573_torchscript.pt2",
75
+ "0.6b": "sapiens_0.6b_goliath_best_goliath_AP_609_torchscript.pt2",
76
+ "1b": "sapiens_1b_goliath_best_goliath_AP_639_torchscript.pt2",
77
  }
78
  DETECTION_CHECKPOINT = os.path.join(CHECKPOINTS_DIR, 'rtmdet_m_8xb32-100e_coco-obj365-person-235e8209.pth')
79
  DETECTION_CONFIG = os.path.join(ASSETS_DIR, 'rtmdet_m_640-8xb32_coco-person_no_nms.py')
assets/checkpoints/{sapiens_0.3b_goliath_best_goliath_AP_575_torchscript.pt2 β†’ sapiens_0.3b_goliath_best_goliath_AP_573_torchscript.pt2} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:21cf7e3e723720d847bee6d3b321bfcdb33268c9f1418d7552552264ae0a5a9b
3
- size 1319579523
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:59ea239bfb217caf76cbeac1836460ef8f9ab5f39aeb22f2577373b62fa25edc
3
+ size 1319574979
assets/checkpoints/{sapiens_0.6b_goliath_best_goliath_AP_600_torchscript.pt2 β†’ sapiens_0.6b_goliath_best_goliath_AP_609_torchscript.pt2} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:dde53a65920832a680ae7d589d437e3335609f656610d2bbe9e731ed04482a0e
3
  size 2645835253
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7d5407351b9f20dddea72cfa7be26783835f58d4d565e3aaf4746cccce9b677a
3
  size 2645835253
assets/checkpoints/{sapiens_1b_goliath_best_goliath_AP_640_torchscript.pt2 β†’ sapiens_1b_goliath_best_goliath_AP_639_torchscript.pt2} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c6218c6be17697157f9e65ee34054a94ab8ca0f637380fa5748c18e04814976e
3
- size 4677162331
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d1077c15e66fb2798893e569f71cb063f0b0778af2a62e3c0233807a351c31f3
3
+ size 4677003899