DCBIA-OrthoLab / SlicerDentalModelSeg

This extension aims to provide a GUI for a deep-learning automated teeth segmentation tool that we developed at the University of North Carolina in Chapel Hill in collaboration with the University of Michigan in Ann Arbor.
Apache License 2.0
44 stars 10 forks source link

BUG : Fix crash importation of librairies in cli #23

Closed HUTIN1 closed 1 year ago

HUTIN1 commented 1 year ago

Pandas, torch, tqdm delete at the beginning of the cli code. This was the source of the error.

The itk libraries have been removed, as they are useless for the CLI.

Torch and tqdm are imported after installation. Avoid running the module a second time, if it is used for the first time.

Pytorch_lightning is also installed if it is missing.

@juanprietob