Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

jgrss/stream #36

Merged
merged 170 commits into from
Nov 9, 2022
Merged

jgrss/stream #36

merged 170 commits into from
Nov 9, 2022

Conversation

jgrss
Copy link
Owner

@jgrss jgrss commented Nov 9, 2022 •
edited
Loading

This PR:

  • Re-configures the model architecture to improve the distance transform predictions
  • Improved prediction performance through pytorch-lightning batch predictions
  • Improves the edge calculations during cultionet create by using geometry boundary
  • Adds improved performance for mean and standard deviation calculations
  • Adds experimental orientation
  • Adds experimental Mask-RCNN
  • Adds a Docker file for CUDA 11.6

@jgrss
updated Docker files
4ac3c8c
@jgrss
create data with masks
33bebe6
@jgrss
cleanup data creation and add orientation
859535f
@jgrss
better data loading
b3deaf5
@jgrss
support masks and orientation
9a9a1c1
@jgrss
changed loss imports
7738877
@jgrss
added mse and huber loss wrappers
90a589e
@jgrss
update prediction layers
07991b8
@jgrss
support orientations
4b8c3d4
@jgrss
added mask rcnn
e02162b
@jgrss
formatting
e05f410
@jgrss
formatting
5456b03
@jgrss
updated args
f1bff62
@jgrss
support serial predictions and additional inference layers
a449f63
@jgrss
formatting
1855cf2
@jgrss
support masks and orientation
7a75704
@jgrss
update tests
6f031ec
@jgrss
created new module for constants
cb0741e
@jgrss
created new module for masks
d5e7755
@jgrss
update tests
1bb5296
@jgrss
removed import
777dd84
@jgrss
flatten tensor
c5a592d
@jgrss
add IoU for masks
54802ac
@jgrss
commented unused
6c642a5
@jgrss
reorganization
6c8120f
@jgrss
resize mask data
76bd618
@jgrss
added trainable layers argument
4ef5b56
@jgrss
added trainable layers argument
01eb0b5
@jgrss
experiment with UNET+++
643f814
@jgrss
formatting
48b18cc
@jgrss jgrss self-assigned this Nov 9, 2022
@jgrss jgrss marked this pull request as ready for review November 9, 2022 19:26
@jgrss jgrss merged commit f2182f1 into main Nov 9, 2022
@jgrss jgrss deleted the jgrss/stream branch November 9, 2022 22:25
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees

@jgrss jgrss

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
1 participant
@jgrss