search

martinsbruveris / tensorflow-image-models

TensorFlow port of PyTorch Image Models (timm) - image models with pretrained weights.
https://tfimm.readthedocs.io/en/latest/
Apache License 2.0
286 stars 25 forks source link

Feat/channels #52

Closed martinsbruveris closed 2 years ago

martinsbruveris commented 2 years ago

Added functionality to change in_channels when creating a model.