Skip to content

Releases

v1.1: held-out region and background models

Choose a tag to compare

@Shafwansafi06 Shafwansafi06 released this 11 Nov 11:54

Four ResNet-50 models (448 px) trained on the held-out cluster splits of Experiments 3 and 4. Same recipe as v1.0: SGD (Nesterov, momentum 0.9, lr 0.01, head lr x10, weight decay 1e-4), cosine schedule, 30 epochs, torchvision ImageNet weights (resnet50.tv_in1k), float16 mixed precision. Best checkpoint by val accuracy; weights only.

File Split Mode Epoch Val Held-out test SHA-256
resnet50_448_geography_full.pth Geography (iNaturalist, 156 species) full image 28 81.94% 62.68% (western N. America, 18,953) 527404323a41a8cc331489c29dedaacd483adcb102b58cee58c67f113ee9e703
resnet50_448_bgsplit_full.pth Background clusters (CUB) full image 25 85.66% 83.96% (ground / grass, 2,350) 9501f27a4e4a7927c7fdf5d5de76e61404a03752564a1011717f136719b43dd7
resnet50_448_bgsplit_crop.pth Background clusters (CUB) bird crop 23 86.93% 85.49% d0449c02294d400aca99b1467b3aa9fb0348a793fbb85545c0b2a2f473ab49b2
resnet50_448_bgsplit_birdshift.pth Background clusters (CUB) BirdShift 30 86.65% 84.30% 810a41f863992b8a70016ee9ec34df878d3d63a111f8c82c7bf81855ec202228

These models are trained on the split manifests in data/splits/, not on CUB's official train/test split, so they should not be scored on the official CUB test set (the background split puts about half of the official test images into training). The best model on the official split is still birdshift_resnet50_448_cub.pth in v1.0 (full-image, 84.55%).

Load: BirdResNet(num_classes=200, pretrained=False).load_state_dict(ck["model_state_dict"]). Each file also stores mode, image_size and manifest. Evaluate with scripts/eval_test.py --manifest <split csv> --splits val test (copy the file to checkpoints/<run>/best.pth). Full-image and BirdShift take Resize(512) + CenterCrop(448); the crop model takes the ground-truth box crop, square-padded and resized to 448.

v1.0: ResNet-50, 84.55% on CUB-200-2011 test

Choose a tag to compare

@Shafwansafi06 Shafwansafi06 released this 09 Nov 12:35

Full-image ResNet-50 at 448 px, fine-tuned on all 5,994 CUB-200-2011 training images.

Top-1 on the official test split: 84.55% (4,899 / 5,794).

Recipe: SGD (Nesterov, momentum 0.9, lr 0.01, head lr x10, weight decay 1e-4), cosine schedule, 50 epochs, torchvision ImageNet weights (resnet50.tv_in1k), Resize(512) + RandomCrop(448) + horizontal flip, float16 mixed precision.

File: birdshift_resnet50_448_cub.pth (weights only)
SHA-256: a647b1bbb444ac05985a3bbc2443e79bfb437ee7e83140e1fdafaa4788c1869c

See the README section "Matching the published ResNet-50 number" for loading code and the full comparison.

Note: this is the full-image model; "birdshift" in the file name is the project name. Held-out split models (including BirdShift) are in v1.1.