apple/ml-cvnets
A PyTorch library by Apple for training computer vision models on classification, detection, segmentation, and foundation model tasks.

Velocity · 7d
+1.2
★ / day
Trend
→steady
star history
CVNets is a computer vision training toolkit that supports standard and mobile-optimized models for tasks including image classification, object detection, semantic segmentation, and foundation models. It includes implementations of EfficientNet, Swin Transformer, ViT, Mask R-CNN, and CLIP, with support for data augmentation techniques like RangeAugment.