Image Classification
Transformers
vision
Inference Endpoints
kamalkraj commited on
Commit
2e606a8
1 Parent(s): 47db3e2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -0
README.md CHANGED
@@ -18,6 +18,10 @@ 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
 
 
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
  import urllib.request
27