Hugging Face's logo

MedicalVision
/
test_remove_2

Object Detection
Transformers
Safetensors
yolos
Inference Endpoints
Model card Files Community
npvinHnivqn commited on
Commit
ad77e67
•
1 Parent(s): a79b010

Upload YolosForObjectDetection

Browse files
Files changed (2) hide show
  1. config.json +15 -15
  2. model.safetensors +1 -1
config.json CHANGED
@@ -38,21 +38,21 @@
38
  "initializer_range": 0.02,
39
  "intermediate_size": 768,
40
  "label2id": {
41
- "LABEL_0": 0,
42
- "LABEL_1": 1,
43
- "LABEL_10": 10,
44
- "LABEL_11": 11,
45
- "LABEL_12": 12,
46
- "LABEL_13": 13,
47
- "LABEL_14": 14,
48
- "LABEL_2": 2,
49
- "LABEL_3": 3,
50
- "LABEL_4": 4,
51
- "LABEL_5": 5,
52
- "LABEL_6": 6,
53
- "LABEL_7": 7,
54
- "LABEL_8": 8,
55
- "LABEL_9": 9
56
  },
57
  "layer_norm_eps": 1e-12,
58
  "model_type": "yolos",
 
38
  "initializer_range": 0.02,
39
  "intermediate_size": 768,
40
  "label2id": {
41
+ "Aortic enlargement": 0,
42
+ "Atelectasis": 1,
43
+ "Calcification": 2,
44
+ "Cardiomegaly": 3,
45
+ "Consolidation": 4,
46
+ "ILD": 5,
47
+ "Infiltration": 6,
48
+ "Lung Opacity": 7,
49
+ "No finding": 14,
50
+ "Nodule/Mass": 8,
51
+ "Other lesion": 9,
52
+ "Pleural effusion": 10,
53
+ "Pleural thickening": 11,
54
+ "Pneumothorax": 12,
55
+ "Pulmonary fibrosis": 13
56
  },
57
  "layer_norm_eps": 1e-12,
58
  "model_type": "yolos",
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4937a42fbf4e3e199c77f13951e6b507ba3260c37ddad93bee35e4f1781b8477
3
  size 25920216
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8f335252ac212cc9d0c14214c3ccbacd48f7e9190a8830f414fb0407cda0f5b6
3
  size 25920216