Skip to content

Releases: mwalmsley/zoobot

Add greyscale encoders

14 May 20:07
1517b20
Compare
Choose a tag to compare

Zoobot now includes pretrained greyscale encoders. FinetuneableZoobot... classes can use these pretrained encoders.

This HuggingFace collection lists all current greyscale encoders.

v2.0

04 Apr 19:26
e6279c1
Compare
Choose a tag to compare

What's Changed

  • New pretrained architectures: ConvNeXT, EfficientNetV2, MaxViT, and more. Each in several sizes. Available on HuggingFace.
  • Reworked finetuning procedure. All these architectures are finetuneable through a common method.
  • Reworked finetuning options. Batch norm finetuning removed. Cosine schedule option added.
  • Reworked finetuning saving/loading. Auto-downloads encoder from HuggingFace.
  • Now supports regression finetuning (as well as multi-class and binary). See pytorch/examples/finetuning
  • Updated timm to 0.9.10, allowing latest model architectures. Previously downloaded checkpoints may not load correctly!
  • (internal until published) GZ Evo v2 now includes Cosmic Dawn (HSC H2O). Significant performance improvement on HSC finetuning. Also now includes GZ UKIDSS (dragged from our archives).
  • Updated pytorch to 2.1.0
  • Added support for webdatasets (only recommended for large-scale distributed training)
  • Improved per-question logging when training from scratch
  • Added option to compile encoder for max speed (not recommended for finetuning, only for pretraining).
  • Deprecates TensorFlow. The CS research community focuses on PyTorch and new frameworks like JAX.

Full Changelog: v1.0.5...v2.0

v1.0.5 - Finetuning Improvements, Max-ViT

13 Nov 20:15
3214941
Compare
Choose a tag to compare

What's Changed

Major improvements to finetuning.

  • Now supports finetuning resnet and maxvit-tiny! This previously only worked as static pretrained models.
  • Now specify num_blocks instead of num_layers, reflecting the network structure (effnet, resnet50, maxvit)
  • Now you can optionally keep batchnorm layers always trainable

Small changes:
Update dependency to use galaxy-datasets==0.15.0
Replace multi-class example with new galaxy-mnist dataset
Fixes #108

The next release will update timm to 0.9ish
Pytorch 2.1.0 already works, and the next release will also reflect that

Full Changelog: v1.0.4...v1.0.5

Color and multi-class support

01 Aug 07:41
ad78db8
Compare
Choose a tag to compare

This minor release adds pretrained color models (i.e. with 3 channels) and also adds support for multi-class finetuning.

What's Changed

Full Changelog: v1.0.3...v1.0.4

JOSS paper release

04 May 11:38
d9e8c2c
Compare
Choose a tag to compare

Minor release marking Zoobot's state on acceptance of the JOSS paper.

What's Changed

Full Changelog: 1.0.2...v1.0.3

v1.0.2 - Lighning v2 and galaxy-datasets updates

29 Mar 09:06
4bb1cbf
Compare
Choose a tag to compare
  • Lightning v2 now correctly uses early stopping
  • galaxy-datasets version bumped to correctly ignore alpha channel of png images

What's Changed

  • Hotfix - add early stopping back, update docs by @mwalmsley in #91

Full Changelog: v1.0.1...1.0.2

v1.0.1 - Lightning v2.0 support

20 Mar 18:56
b9f9812
Compare
Choose a tag to compare

Small incremental release adjusting the LightningModule hooks used in order to support Lightning v2. Lightning v2.0.0 (currently latest) is now the minimum required version.

What's Changed

Full Changelog: v1.0.0...v1.0.1

v1 release

10 Mar 17:01
Compare
Choose a tag to compare

This release completely rewrites the API and documentation based on user feedback during the one-year beta period.

Major changes include:

  • New API for finetuning
  • Shift to PyTorch/Lightning
  • Support for all timm models
  • Docs refocused on finetuning
  • Pretrained model library added
  • Data loading refactored to albumentations and mwalmsley/galaxy-datasets

Documentation is here. Installation and quickstart are on the README.

Thank you all for the help!

What's Changed

New Contributors

Full Changelog: v0.0.3...v1.0.0

Docs pre-release

25 Apr 13:25
Compare
Choose a tag to compare
Docs pre-release Pre-release
Pre-release

Improved documentation and refactored train API (pytorch).

Awaiting results from several segmentation experiments ahead of public release (inc pytorch version).

v0.0.2

04 Oct 13:31
1d3511e
Compare
Choose a tag to compare
v0.0.2 Pre-release
Pre-release

Polish throughout and add new features. See #2 for a full description.