qiuyuan666

> Go to the function dice_core_metric of losses.py file and uncomment the following lines of code and comment out the following. > def dice_core_metric(y_true, y_pred): > ##computes the dice for...

Hello, I used your solution(change self.model) and ran into this problem as well:No such file or directory: '*******/pretrained_weights/ResUnet.epoch_02.hdf5.json' I don't understand you said "The json file can be produced by...

Hello! Mehrdad-Noori. Thanks for your code! I want to train my own data, how do I put my own dataset into the folder 'data'? What are the data format requirements?...

I have the same problem, Can anyone help with this?

非常感谢！我会去看看您发的百度的那个医学分割实例。 不过我还是想知道的是如何训练自己的model，比如，您在readme里面已经给出了human segmentation model（u2net_human_seg.pth ）和针对 salient object detection的 pre-trained model u2net.pth。那么如果我不用您给出的预训练模型，想训练自己的model，我要怎么做？

不好意思，我好没有找到方法，你是怎么训练自己的模型的？ ------------------ 原始邮件 ------------------ 发件人: "xuebinqin/U-2-Net" ***@***.***>; 发送时间: 2021年9月3日(星期五) 下午3:39 ***@***.***>; ***@***.******@***.***>; 主题: Re: [xuebinqin/U-2-Net] 训练自己的模型，用于医学图像分割 (#232) 非常感谢！我会去看看您发的百度的那个医学分割实例。 不过我还是想知道的是如何训练自己的model，比如，您在readme里面已经给出了human segmentation model（u2net_human_seg.pth ）和针对 salient object detection的 pre-trained model u2net.pth。那么如果我不用您给出的预训练模型，想训练自己的model，我要怎么做？ 你找到方法了吗？我做法了加载了他们预训练的pth，然后让模型在自己的数据集上跑。试过从头到尾自己训练的模型，效果没加载预训练的好。 — You are receiving this...

您好，又来打扰您啦~ 运行train_model.py遇到的问题： 1.运行到第152行，get_training_and_validation_generators( )函数时会进入generator.py执行该函数，执行完毕后发现读取的datafile是生成的data"文件夹下的data.h5文件，显示training_list, validation_list，num_patches_training，num_patches_val都是0，这是因为根本没有读取到数据吧？ 在运行之前没有单独运行process.py，没有做预处理将数据存在对应文件夹，所以generator.py里的函数就没读到数据吗？ 2.继续运行到第177行进入train_model( )函数后就一直在这个函数里不动了，具体显示如下： WARNING:tensorflow:From /home/anaconda3/envs/tensorflow/lib/python3.6/site-packages/keras/backend/tensorflow_backend.py:3445: calling dropout (from tensorflow.python.ops.nn_ops) with keep_prob is deprecated and will be removed in a future version. Instructions for updating: Please...

您好，我运行了预处理的代码，预处理后的数据存放目录和您的original_tree对比后是一样的。然后我重新运行了train_model.py，但是还是training_list, validation_list，num_patches_training，num_patches_val都是0，运行后结果如下： ‘’‘ totalMemory: 7.79GiB freeMemory: 7.35GiB 2020-10-09 15:34:57.603044: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1512] Adding visible gpu devices: 0 2020-10-09 15:34:57.604181: I tensorflow/core/common_runtime/gpu/gpu_device.cc:984] Device interconnect StreamExecutor with strength 1 edge matrix: 2020-10-09 15:34:57.604200:...

I have the same issue. Did you manage to solve it? Could you please guide me? @kmbmjn @Mehrdad-Noori @

Where can I download "vit_base_patch16_224" and "jx_vit_base_p16_224-80ecf9dd.pth"? Looking forward to and thank you for your reply. @MengzhangLI