search

jcreinhold / tiramisu-brulee

A 2D and 3D PyTorch implementation of the Tiramisu CNN
Other
5 stars 2 forks source link

Make training and prediction framework model agnostic #3

Closed jcreinhold closed 3 years ago

jcreinhold commented 3 years ago

The current lesionseg framework integrates the Tiramisu model into training and prediction framework. It would be preferable to separate the two so you can import the training/prediction and drop-in your own model.

jcreinhold commented 3 years ago

Completed with 5d70d784c51d49ef8c83d72b39be49ba32fd1c49