Image Classification
Transformers
vision
Inference Endpoints
kamalkraj commited on
Commit
e9e658f
1 Parent(s): 079472a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -0
README.md CHANGED
@@ -18,7 +18,12 @@ A DenseNet is a type of convolutional neural network that utilises dense connect
18
 
19
  Here is how to use this model to classify an image of xray:
20
 
 
 
 
 
21
  ```python
 
22
  import urllib.request
23
 
24
  import skimage
 
18
 
19
  Here is how to use this model to classify an image of xray:
20
 
21
+ Note: Each pretrained model has 18 outputs. The `all` model has every output trained. However, for the other weights some targets are not trained and will predict randomly becuase they do not exist in the training dataset. The only valid outputs are listed in the field `{dataset}.pathologies` on the dataset that corresponds to the weights.
22
+
23
+
24
+ Benchmarks of the modes are here: [BENCHMARKS.md](https://github.com/mlmed/torchxrayvision/blob/master/BENCHMARKS.md)
25
  ```python
26
+
27
  import urllib.request
28
 
29
  import skimage