Hugging Face's logo

RishiDarkDevil
/
ai-image-det-resnet152

Feature Extraction
Transformers
PyTorch
English
resnet
classification
computer vision
custom_code
Model card Files Community
2

Upload feature extractor

#2
by RishiDarkDevil - opened
base: refs/heads/main
←
from: refs/pr/2
Discussion Files changed
+4
-3
Files changed (2) hide show
  1. config.json +2 -1
  2. pytorch_model.bin +2 -2
config.json CHANGED
@@ -1,9 +1,10 @@
1
  {
2
  "architectures": [
3
- "ResnetModelForImageClassification"
4
  ],
5
  "auto_map": {
6
  "AutoConfig": "configuration_resnet.ResnetFeatureExtractorConfig",
 
7
  "AutoModelForImageClassification": "model_resnet.ResnetModelForImageClassification"
8
  },
9
  "model_type": "resnet",
 
1
  {
2
  "architectures": [
3
+ "ResnetFeatureExtractor"
4
  ],
5
  "auto_map": {
6
  "AutoConfig": "configuration_resnet.ResnetFeatureExtractorConfig",
7
+ "AutoModel": "model_resnet.ResnetFeatureExtractor",
8
  "AutoModelForImageClassification": "model_resnet.ResnetModelForImageClassification"
9
  },
10
  "model_type": "resnet",
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0b809790a34c7cf9a393c984d6ccdf0049dd07b03a37469a40c1527593da1b43
3
- size 264131
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:54a04776d4bf8dda8b4beebf6019f30567bee86c9a1e89b5f04e81f8e5a58392
3
+ size 233468565