Skip to content

3D augmentation and 2.1D U-Net - #27

Merged
mattersoflight merged 35 commits into
mainfrom
3d-augmentation
Aug 12, 2023
Merged

mattersoflight merged 35 commits into
mainfrom
3d-augmentation

Conversation

@ziw-liu

@ziw-liu ziw-liu commented Aug 9, 2023

Copy link
Copy Markdown
Contributor

Implemented the Z-scaling augmentation in #23.

To test the large input stack depth I also made a new hybrid network using a 3D-to-2D projection stem followed by any off-the-shelf ResNet-like feature encoder (from timm, default to convnextv2_femto) and a plain 2D U-Net decoder. This is temporarily named 2.1D U-Net. We can decide on whether to keep it after seeing some training results.

@ziw-liu ziw-liu added the documentation Improvements or additions to documentation label Aug 10, 2023
ziw-liu and others added 9 commits August 10, 2023 11:41
* readme + dependencies tested with python 3.10

* match package description with README

* cellpose (metrics -> required)

* misc formatting and language edits

* fix optional dependency on cellpose

---------

Co-authored-by: Ziwen Liu <ziwen.liu@czbiohub.org>
@mattersoflight
mattersoflight marked this pull request as ready for review August 12, 2023 14:06
@mattersoflight
mattersoflight merged commit f9b4e16 into main Aug 12, 2023
@ziw-liu
ziw-liu deleted the 3d-augmentation branch August 12, 2023 21:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

documentation Improvements or additions to documentation

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants