Spaces:
Sleeping
Sleeping
Commit
·
3cb2603
1
Parent(s):
de89970
Changed model from Vit Xsmall clip to resnet50_clip_gap. Changed descriptioin and .onnx file
Browse files
.gitignore
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
.ipynb_checkpoints/
|
app.py
CHANGED
@@ -10,7 +10,7 @@ input_size = [1, 3, 224, 224]
|
|
10 |
img_size = input_size[-1]
|
11 |
|
12 |
title = "Gender Vision mini"
|
13 |
-
description = "An
|
14 |
article = "Through bunch of fine tuning and experiments. !REMEMBER! This model can be wrong."
|
15 |
|
16 |
def predict(img):
|
|
|
10 |
img_size = input_size[-1]
|
11 |
|
12 |
title = "Gender Vision mini"
|
13 |
+
description = "An resnet50_clip_gap based model(fine tuned with Custom dataset : around 800 train images & 200 test iamges) F1 Score : 100%(1.00) with the custom test dataset. Optimized with ONNX(around 1.7 times faster than PyTorch version on cpu)"
|
14 |
article = "Through bunch of fine tuning and experiments. !REMEMBER! This model can be wrong."
|
15 |
|
16 |
def predict(img):
|
deploying model/{vit_xsmall_patch16_clip_224(trainble_0.00) (F1 Score 96.83%, Loss 0.229548).onnx → resnet50_clip_gap(trainble_0.00) (F1 Score 100.00%, Loss 0.149292).onnx}
RENAMED
@@ -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:b5bbf7a95eda37fe6b85764eac8d0796221e07f92c7cfcff91ee779d5cd3b833
|
3 |
+
size 94052414
|