Skip to content

Commit

Permalink
feat: 🚀 D-FINE results are added (COCO only models)
Browse files Browse the repository at this point in the history
Signed-off-by: Onuralp SEZER <[email protected]>
  • Loading branch information
onuralpszr committed Nov 30, 2024
1 parent 77c002d commit d8958bb
Show file tree
Hide file tree
Showing 13 changed files with 1,705 additions and 465 deletions.
2 changes: 2 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,8 @@ data/*

# yolov9 repo
**/yolov9-repo
# D-FINE repo
**/D-FINE
# mmyolo-weights
**/mmyolo-weights

Expand Down
1 change: 1 addition & 0 deletions build_static_site.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,3 +29,4 @@
f.write(js_results)

print("Results aggregated and saved to static/aggregate_results.js")
print("Results aggregated and saved to static/aggregate_results.json")
Empty file.
10 changes: 10 additions & 0 deletions models/object_detection/d-fine/requirements.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
supervision @ git+https://github.com/roboflow/supervision.git@develop
torch>=2.0.1
torchvision>=0.15.2
faster-coco-eval>=1.6.5
PyYAML
tensorboard
scipy
calflops
transformers
tqdm
68 changes: 68 additions & 0 deletions models/object_detection/d-fine/results_D‑FINE‑L.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,68 @@
{
"metadata": {
"model": "D\u2011FINE\u2011L",
"license": "Apache-2.0",
"run_parameters": {
"imgsz": 640,
"conf": 0.001
},
"param_count": 30724195,
"run_date": "2024-11-30T15:31:29.563018+00:00"
},
"map50_95": 0.5219932815159587,
"map50": 0.6952979708509999,
"map75": 0.5653622772704325,
"small_objects": {
"map50_95": 0.2477807526608342,
"map50": 0.39281358142909667,
"map75": 0.2682646010589643
},
"medium_objects": {
"map50_95": 0.4757793179580914,
"map50": 0.6506829538892683,
"map75": 0.5321307963239714
},
"large_objects": {
"map50_95": 0.6679411840312329,
"map50": 0.7990835690433563,
"map75": 0.7183590809069729
},
"iou_thresholds": [
0.5,
0.55,
0.6,
0.65,
0.7,
0.75,
0.8,
0.85,
0.8999999999999999,
0.95
],
"f1_50": 0.055112414649932884,
"f1_75": 0.045577933029039445,
"f1_small_objects": {
"f1_50": 0.0538862414751944,
"f1_75": 0.03414610933748649
},
"f1_medium_objects": {
"f1_50": 0.09175005181275003,
"f1_75": 0.07934473563523216
},
"f1_large_objects": {
"f1_50": 0.08174917885040568,
"f1_75": 0.07779861089961485
},
"f1_iou_thresholds": [
0.5,
0.55,
0.6,
0.65,
0.7,
0.75,
0.8,
0.85,
0.8999999999999999,
0.95
]
}
68 changes: 68 additions & 0 deletions models/object_detection/d-fine/results_D‑FINE‑M.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,68 @@
{
"metadata": {
"model": "D\u2011FINE\u2011M",
"license": "Apache-2.0",
"run_parameters": {
"imgsz": 640,
"conf": 0.001
},
"param_count": 19248189,
"run_date": "2024-11-30T15:27:16.048157+00:00"
},
"map50_95": 0.5058476134984426,
"map50": 0.6784489386840735,
"map75": 0.5458102809213676,
"small_objects": {
"map50_95": 0.22101506894321837,
"map50": 0.35919423860167804,
"map75": 0.23411438906708412
},
"medium_objects": {
"map50_95": 0.4625887037623695,
"map50": 0.6360438151828612,
"map75": 0.5153654469224979
},
"large_objects": {
"map50_95": 0.6525501696792834,
"map50": 0.7865990282572766,
"map75": 0.7028242508843852
},
"iou_thresholds": [
0.5,
0.55,
0.6,
0.65,
0.7,
0.75,
0.8,
0.85,
0.8999999999999999,
0.95
],
"f1_50": 0.05439085557470057,
"f1_75": 0.04462990245971832,
"f1_small_objects": {
"f1_50": 0.054219329911978476,
"f1_75": 0.033933781321204996
},
"f1_medium_objects": {
"f1_50": 0.09002828069737208,
"f1_75": 0.07714710066070882
},
"f1_large_objects": {
"f1_50": 0.07786530182363084,
"f1_75": 0.07370602951705649
},
"f1_iou_thresholds": [
0.5,
0.55,
0.6,
0.65,
0.7,
0.75,
0.8,
0.85,
0.8999999999999999,
0.95
]
}
68 changes: 68 additions & 0 deletions models/object_detection/d-fine/results_D‑FINE‑N.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,68 @@
{
"metadata": {
"model": "D\u2011FINE\u2011N",
"license": "Apache-2.0",
"run_parameters": {
"imgsz": 640,
"conf": 0.001
},
"param_count": 3753799,
"run_date": "2024-11-30T14:53:36.827013+00:00"
},
"map50_95": 0.41166844751896103,
"map50": 0.5804468688476232,
"map75": 0.44032133935943624,
"small_objects": {
"map50_95": 0.14772166810764642,
"map50": 0.25995371220767033,
"map75": 0.15252458599054255
},
"medium_objects": {
"map50_95": 0.35285141528183744,
"map50": 0.5253203750965515,
"map75": 0.3862506701611196
},
"large_objects": {
"map50_95": 0.5746650046013381,
"map50": 0.7198294931560041,
"map75": 0.6225356620937137
},
"iou_thresholds": [
0.5,
0.55,
0.6,
0.65,
0.7,
0.75,
0.8,
0.85,
0.8999999999999999,
0.95
],
"f1_50": 0.05058755898129688,
"f1_75": 0.037554961513929404,
"f1_small_objects": {
"f1_50": 0.04851364560060535,
"f1_75": 0.021423510193421754
},
"f1_medium_objects": {
"f1_50": 0.0776416142825443,
"f1_75": 0.06009583714479435
},
"f1_large_objects": {
"f1_50": 0.07216459506892758,
"f1_75": 0.06711776075482231
},
"f1_iou_thresholds": [
0.5,
0.55,
0.6,
0.65,
0.7,
0.75,
0.8,
0.85,
0.8999999999999999,
0.95
]
}
68 changes: 68 additions & 0 deletions models/object_detection/d-fine/results_D‑FINE‑S.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,68 @@
{
"metadata": {
"model": "D\u2011FINE\u2011S",
"license": "Apache-2.0",
"run_parameters": {
"imgsz": 640,
"conf": 0.001
},
"param_count": 10237491,
"run_date": "2024-11-30T15:25:06.497738+00:00"
},
"map50_95": 0.4667940572641105,
"map50": 0.6315233422975106,
"map75": 0.5075257211881392,
"small_objects": {
"map50_95": 0.2023262851372202,
"map50": 0.32968181064456004,
"map75": 0.21768401657888195
},
"medium_objects": {
"map50_95": 0.41703368286419307,
"map50": 0.5832924172471692,
"map75": 0.4675721643466931
},
"large_objects": {
"map50_95": 0.612227433009285,
"map50": 0.7454489601512843,
"map75": 0.6625822766791043
},
"iou_thresholds": [
0.5,
0.55,
0.6,
0.65,
0.7,
0.75,
0.8,
0.85,
0.8999999999999999,
0.95
],
"f1_50": 0.053318704601921355,
"f1_75": 0.04302370379707862,
"f1_small_objects": {
"f1_50": 0.05341393465811958,
"f1_75": 0.03234633429835207
},
"f1_medium_objects": {
"f1_50": 0.08474685091911431,
"f1_75": 0.0709119555892223
},
"f1_large_objects": {
"f1_50": 0.07746434221415874,
"f1_75": 0.07309819351528968
},
"f1_iou_thresholds": [
0.5,
0.55,
0.6,
0.65,
0.7,
0.75,
0.8,
0.85,
0.8999999999999999,
0.95
]
}
68 changes: 68 additions & 0 deletions models/object_detection/d-fine/results_D‑FINE‑X.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,68 @@
{
"metadata": {
"model": "D\u2011FINE\u2011X",
"license": "Apache-2.0",
"run_parameters": {
"imgsz": 640,
"conf": 0.001
},
"param_count": 61601635,
"run_date": "2024-11-30T15:34:31.249037+00:00"
},
"map50_95": 0.5412884800986779,
"map50": 0.7166391180688009,
"map75": 0.5843090799180614,
"small_objects": {
"map50_95": 0.26577391016604135,
"map50": 0.4204213608890482,
"map75": 0.2849821329125042
},
"medium_objects": {
"map50_95": 0.49187707994641156,
"map50": 0.667776251420199,
"map75": 0.5506275489522772
},
"large_objects": {
"map50_95": 0.6894573901354175,
"map50": 0.8207794964547676,
"map75": 0.7393501736230832
},
"iou_thresholds": [
0.5,
0.55,
0.6,
0.65,
0.7,
0.75,
0.8,
0.85,
0.8999999999999999,
0.95
],
"f1_50": 0.05502261232306097,
"f1_75": 0.045594707812090164,
"f1_small_objects": {
"f1_50": 0.05357001873043499,
"f1_75": 0.033972001296026455
},
"f1_medium_objects": {
"f1_50": 0.09390118651490137,
"f1_75": 0.08149773985318125
},
"f1_large_objects": {
"f1_50": 0.08145177470815365,
"f1_75": 0.07772815532345857
},
"f1_iou_thresholds": [
0.5,
0.55,
0.6,
0.65,
0.7,
0.75,
0.8,
0.85,
0.8999999999999999,
0.95
]
}
Loading

0 comments on commit d8958bb

Please sign in to comment.