Spaces:
Running
Running
dmitryhits
commited on
Commit
·
d2def97
1
Parent(s):
3e28d8a
changes article
Browse files
app.py
CHANGED
@@ -19,12 +19,13 @@ def predict(img):
|
|
19 |
return {labels[i]: float(probs[i]) for i in range(len(labels))}
|
20 |
|
21 |
# %% ../weed_classifier.ipynb 5
|
22 |
-
article = """<div>This model was trained on the dataset contains 5,539 images of crop and weed seedlings.
|
23 |
The images are grouped into 12 classes.
|
24 |
These classes represent common plant species in Danish agriculture at different growth stages.
|
25 |
A pretrained ResNet34 model was fine-tuned using fastai vision library. The error rate achived on the validation dataset is 2.5% after 4 epochs.
|
26 |
Currently the model does not generalize well on the out-of-domain images. Probably because of the specificity of the dataset.
|
27 |
-
The images in the dataset are of a single plant taken from the top and on a single background of a grit surface. Some images have labels that might be unique to the particular plant.
|
|
|
28 |
<p style='text-align: center'><a href='https://tmabraham.github.io/blog/gradio_hf_spaces_tutorial' target='_blank'>Blog post</a></p>
|
29 |
</div>"""
|
30 |
|
|
|
19 |
return {labels[i]: float(probs[i]) for i in range(len(labels))}
|
20 |
|
21 |
# %% ../weed_classifier.ipynb 5
|
22 |
+
article = """<div>This model was trained on the dataset that contains 5,539 images of crop and weed seedlings.
|
23 |
The images are grouped into 12 classes.
|
24 |
These classes represent common plant species in Danish agriculture at different growth stages.
|
25 |
A pretrained ResNet34 model was fine-tuned using fastai vision library. The error rate achived on the validation dataset is 2.5% after 4 epochs.
|
26 |
Currently the model does not generalize well on the out-of-domain images. Probably because of the specificity of the dataset.
|
27 |
+
The images in the dataset are of a single plant taken from the top and on a single background of a grit surface. Some images have barcode labels that might be unique to the particular plant.
|
28 |
+
The last two images in the examples are not from the dataset. They are getting wrongly classified as <b>Common Chickweed</b> even though they are <b>Small-flowered Cranesbill</b> (compare with the second image in the examples)
|
29 |
<p style='text-align: center'><a href='https://tmabraham.github.io/blog/gradio_hf_spaces_tutorial' target='_blank'>Blog post</a></p>
|
30 |
</div>"""
|
31 |
|