rawalkhirodkar commited on
Commit
64ef3c8
1 Parent(s): 28c256d

Add sapiens-0.6b

Browse files
app.py CHANGED
@@ -72,6 +72,7 @@ class Config:
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
  "1b": "sapiens_1b_goliath_best_goliath_AP_640_torchscript.pt2",
76
  }
77
  DETECTION_CHECKPOINT = os.path.join(CHECKPOINTS_DIR, 'rtmdet_m_8xb32-100e_coco-obj365-person-235e8209.pth')
 
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')
assets/checkpoints/sapiens_0.6b_goliath_best_goliath_AP_600_torchscript.pt2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dde53a65920832a680ae7d589d437e3335609f656610d2bbe9e731ed04482a0e
3
+ size 2645835253