sovit-123 / fasterrcnn-pytorch-training-pipeline

PyTorch Faster R-CNN Object Detection on Custom Dataset
MIT License
221 stars 73 forks source link

what gives me best accuracy of theses models for small object #90

Closed VYRION-Ai closed 1 year ago

VYRION-Ai commented 1 year ago

what gives me best accuracy of theses models for small object

''[ 'fasterrcnn_convnext_small', 'fasterrcnn_convnext_tiny', 'fasterrcnn_custom_resnet', 'fasterrcnn_darknet', 'fasterrcnn_efficientnet_b0', 'fasterrcnn_efficientnet_b4', 'fasterrcnn_mbv3_small_nano_head', 'fasterrcnn_mbv3_large', 'fasterrcnn_mini_darknet_nano_head', 'fasterrcnn_mini_darknet', 'fasterrcnn_mini_squeezenet1_1_small_head', 'fasterrcnn_mini_squeezenet1_1_tiny_head', 'fasterrcnn_mobilenetv3_large_320_fpn', # Torchvision COCO pretrained 'fasterrcnn_mobilenetv3_large_fpn', # Torchvision COCO pretrained 'fasterrcnn_nano', 'fasterrcnn_resnet18', 'fasterrcnn_resnet50_fpn_v2', # Torchvision COCO pretrained 'fasterrcnn_resnet50_fpn', # Torchvision COCO pretrained 'fasterrcnn_resnet101', 'fasterrcnn_resnet152', 'fasterrcnn_squeezenet1_0', 'fasterrcnn_squeezenet1_1_small_head', 'fasterrcnn_squeezenet1_1', 'fasterrcnn_vitdet', 'fasterrcnn_vitdet_tiny', 'fasterrcnn_mobilevit_xxs', 'fasterrcnn_regnet_y_400mf' ]''

sovit-123 commented 1 year ago

Hi @VYRION-Ai If you are looking for the best accuracy and FPS is not a concern, then fasterrcnn_resnet50_fpn_v2 works best.

VYRION-Ai commented 1 year ago

thank you