Pytorch-Unet3D-single_channel icon indicating copy to clipboard operation
Pytorch-Unet3D-single_channel copied to clipboard

training set question

Open ljljlj02 opened this issue 4 years ago • 3 comments

Hi, I am not understand about the data set setting of your program in the process of running. May I ask what data set this program applies to?Why do data_1 include A images and B labels in your training set?Shouldn't it be A images and A labels? Also, Can it run under Win10? Any help will be greatly appreciated

ljljlj02 avatar Jun 28 '20 03:06 ljljlj02

It's a general script for segmentation.

  • I used it for carotid artery segmentation in this case. In this repository https://github.com/davidiommi/3D_Prostate_segmentation_pytorch I used it with the Promise2012 prostate dataset ( which is public).
  • With data_1, data_2 I meant the patients list and A= image, B= label for each patient case. In the prostate repository you can read it explicitly. ( I modified the Read.me file to make it clear now).
  • I did run the inference on my Win 10 with Pycharm, but the training on Linux. I guess it won't be a problem doing both with Win 10.

davidiommi avatar Jun 29 '20 12:06 davidiommi

Interesting. I'll try it. Thank you!

ljljlj02 avatar Jun 29 '20 12:06 ljljlj02

Why is it that when I train my model on my computer it automatically reboots after two runs?I would like to ask you if there is a similar situation.I only have a 2080ti GPU and a Ubuntu18.04 system.Is the configuration insufficient?

chenjingyi-star avatar Jul 08 '20 07:07 chenjingyi-star