Mohammad Ahangar

Results 2 comments of Mohammad Ahangar

As you are going to use pre-trained DETR from transformers , you can import it like this: from transformers import DetrImageProcessor image_processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50") Hope this helps!

> After running this block, I am getting "Name Error image_processor is not defined" > > TRAIN_DATASET = CocoDetection(image_directory_path=TRAIN_DIRECTORY, image_processor=image_processor, train=True) VAL_DATASET = CocoDetection(image_directory_path=VAL_DIRECTORY, image_processor=image_processor, train=False) TEST_DATASET = CocoDetection(image_directory_path=TEST_DIRECTORY, image_processor=image_processor,...