.gitattributes

@@ -14,3 +14,4 @@
 *.pb filter=lfs diff=lfs merge=lfs -text
 *.pt filter=lfs diff=lfs merge=lfs -text
 *.pth filter=lfs diff=lfs merge=lfs -text
+model.safetensors filter=lfs diff=lfs merge=lfs -text

README.md

@@ -45,8 +45,9 @@ import requests
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)
 
-processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-101")
+# you can specify the revision tag if you don't want the timm dependency
-model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-101")
+processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-101", revision="no_timm")
+model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-101", revision="no_timm")
 
 inputs = processor(images=image, return_tensors="pt")
 outputs = model(**inputs)

preprocessor_config.json

@@ -1,7 +1,5 @@
 {
   "do_normalize": true,
-  "do_pad": true,
-  "do_rescale": true,
   "do_resize": true,
   "format": "coco_detection",
   "image_mean": [
@@ -9,14 +7,11 @@
     0.456,
     0.406
   ],
-  "image_processor_type": "DetrImageProcessor",
   "image_std": [
     0.229,
     0.224,
     0.225
   ],
-  "resample": 2,
-  "rescale_factor": 0.00392156862745098,
   "size": {
     "longest_edge": 1333,
     "shortest_edge": 800