dmitryhits commited on
Commit
d2def97
1 Parent(s): 3e28d8a

changes article

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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