Stylized-ImageNet
Investigate the method of stylizing ImageNet to increase shape bias in CNNs for better performance. This project offers detailed tools for dataset transformation and model evaluation, supporting both PyTorch and TensorFlow frameworks. Generate in-depth reports to understand model behavior with stylized data and out-of-distribution performance. With a convenient Docker setup, this resource aids in creating Stylized-ImageNet and includes pre-trained CNNs for practical demonstration. Ideal for research focused on enhancing generalization in machine learning.
