锟斤拷锟斤拷位锟矫ｏ拷锟斤拷页 > 锟斤拷锟斤拷锟教筹拷 > 锟教筹拷 > 锟絅锟节匡拷锟斤拷全锟斤拷锟斤拷之 MindSpore锟斤拷为锟斤拷模锟斤拷训锟斤拷

锟絅锟节匡拷锟斤拷全锟斤拷锟斤拷之 MindSpore锟斤拷为锟斤拷模锟斤拷训锟斤拷

锟斤拷源锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷|锟斤拷时锟戒：2024-05-26 08:53:46 |锟斤拷锟侥讹拷锟斤拷110锟斤拷|锟斤拷锟斤拷签锟斤拷全 S in 锟斤拷锟斤拷锟斤拷为锟斤拷 |锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷

学锟斤拷锟斤拷伟锟阶帮拷锟斤拷没锟轿拷锟組odelArts锟斤拷锟斤拷锟斤拷锟斤拷Atlas 200I DK A2锟斤拷锟斤拷锟斤拷通一锟斤拷训锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷全锟斤拷锟斤拷思路锟斤拷 > 锟节憋拷篇锟铰ｏ拷锟斤拷锟斤拷锟斤拷锟饺匡拷始训锟斤拷锟阶段ｏ拷

前锟斤拷

学锟斤拷锟斤拷伟锟阶帮拷锟斤拷没锟轿拷锟組odelArts锟斤拷锟斤拷锟斤拷锟斤拷Atlas 200I DK A2锟斤拷
锟斤拷锟斤拷通一锟斤拷训锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷全锟斤拷锟斤拷思路锟斤拷

锟节憋拷篇锟铰ｏ拷锟斤拷锟斤拷锟斤拷锟饺匡拷始训锟斤拷锟阶段ｏ拷

训锟斤拷锟阶讹拷

A. 锟斤拷锟斤拷锟筋建

MindSpore 锟斤拷为锟斤拷模锟斤拷训锟斤拷

Step1 锟斤拷锟斤拷OBS锟斤拷锟斤拷锟侥硷拷

锟斤拷录锟斤拷为锟斤拷 -> 锟斤拷锟斤拷台 -> 锟斤拷嗟硷拷锟斤拷锟窖★拷瘛岸锟斤拷锟芥储锟斤拷锟斤拷 OBS锟斤拷 ->
锟斤拷锟斤拷嗟硷拷锟斤拷锟窖★拷锟酵帮拷斜锟斤拷锟� -> 锟斤拷锟斤拷锟斤拷锟较角★拷锟斤拷锟斤拷桶锟斤拷
锟斤拷锟斤拷图锟斤拷示锟斤拷
锟斤拷锟斤拷锟斤拷斜锟斤拷械摹锟斤拷锟斤拷锟斤拷募锟较低筹拷锟� -> 锟斤拷锟斤拷锟斤拷锟较角★拷锟斤拷锟斤拷锟斤拷锟斤拷锟侥硷拷系统锟斤拷锟斤拷
锟斤拷锟斤拷图锟斤拷示锟斤拷

锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟矫ｏ拷

锟斤拷要锟斤拷锟斤拷锟斤拷息锟斤拷锟铰ｏ拷锟斤拷锟斤拷锟斤拷锟斤拷锟诫保锟斤拷默锟斤拷锟斤拷锟斤拷
锟斤拷锟斤拷 锟斤拷选锟斤拷 锟斤拷锟斤拷-锟斤拷锟斤拷锟斤拷 锟斤拷
锟侥硷拷系统锟斤拷锟狡ｏ拷锟皆讹拷锟藉，锟斤拷锟斤拷使锟斤拷modelarts0009
锟斤拷锟斤拷使锟斤拷modelarts锟斤拷为锟侥硷拷系统前缀锟斤拷注锟斤拷锟斤拷锟斤拷为全锟斤拷唯一锟斤拷
锟斤拷锟斤拷锟斤拷锟斤拷娲拷锟斤拷锟� 锟斤拷选锟斤拷 锟斤拷AZ锟芥储 锟斤拷
锟斤拷锟斤拷 锟斤拷选锟斤拷 私锟斤拷 锟斤拷

Step2 锟较达拷锟斤拷锟斤拷锟侥硷拷锟斤拷OBS锟斤拷锟斤拷锟侥硷拷系统

锟斤拷锟斤拷汛锟斤拷锟斤拷牟锟斤拷锟斤拷募锟较低� -> 锟斤拷锟斤拷锟斤拷陆锟斤拷募锟斤拷小锟�
锟斤拷锟斤拷锟侥硷拷锟叫碉拷锟斤拷锟狡ｏ拷锟斤拷锟斤拷锟斤拷锟斤拷为input
锟斤拷锟斤拷锟斤拷募锟斤拷锟斤拷锟� -> 锟斤拷锟斤拷锟斤拷洗锟斤拷募锟斤拷锟斤拷锟�
锟斤拷准锟斤拷锟矫碉拷锟斤拷目锟斤拷锟斤拷锟侥硷拷压锟斤拷锟斤拷锟较达拷锟斤拷锟斤拷OBS锟叫★拷

Step3 锟斤拷锟斤拷ModelArts锟斤拷锟斤拷Notebook锟斤拷袒锟斤拷锟�

锟节★拷全锟斤拷锟斤拷锟矫★拷页锟斤拷榭达拷欠锟斤拷丫锟斤拷锟斤拷锟斤拷锟饺拷锟斤拷锟斤拷锟組odelArts锟斤拷锟斤拷OBS锟斤拷
锟斤拷录锟斤拷为锟斤拷 -> 锟斤拷锟斤拷台 -> 锟斤拷嗟硷拷锟斤拷锟窖★拷锟組odelArts锟斤拷 -> 锟斤拷锟斤拷嗟硷拷锟斤拷锟窖★拷锟饺拷锟斤拷锟斤拷谩锟� -> 锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷权锟斤拷
锟阶达拷使锟斤拷ModelArts锟斤拷直锟斤拷选锟斤拷锟斤拷锟斤拷委锟叫★拷锟叫的★拷锟斤拷通锟矫伙拷锟斤拷权锟斤拷
锟斤拷录锟斤拷为锟斤拷 -> 锟斤拷锟斤拷台 -> 锟斤拷嗟硷拷锟斤拷锟窖★拷锟組odelArts锟斤拷 -> 锟斤拷锟斤拷嗟硷拷锟斤拷锟窖★拷瘛翱锟斤拷锟斤拷锟斤拷锟斤拷锟�-> 锟斤拷Notebook锟斤拷 -> 锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟�
锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟矫ｏ拷

锟斤拷要锟斤拷锟斤拷锟斤拷息锟斤拷锟铰ｏ拷锟斤拷锟斤拷锟斤拷锟斤拷锟诫保锟斤拷默锟斤拷锟斤拷锟斤拷
锟斤拷锟狡ｏ拷锟皆讹拷锟藉，锟斤拷锟斤拷使锟斤拷notebook-test
锟皆讹拷停止锟斤拷锟斤拷锟斤拷选锟今，憋拷锟斤拷选锟斤拷4小时
锟斤拷锟斤拷 锟斤拷选锟今“癸拷锟斤拷锟斤拷锟今”ｏ拷锟斤拷选锟斤拷 mindspore_1.10.0-cann_6.0.1-py_3.7-euler_2.8.3 锟斤拷
锟斤拷源锟斤拷锟酵ｏ拷选锟今“癸拷锟斤拷锟斤拷源锟截★拷
锟斤拷锟教癸拷锟� 锟斤拷使锟斤拷 50 GB

Step4 为Notebook锟斤拷袒锟斤拷锟斤拷锟斤拷锟窖碉拷锟斤拷锥锟斤拷锟侥匡拷锟斤拷锟斤拷募锟�

锟斤拷锟斤拷汛锟斤拷锟斤拷锟絅otebook -> 锟斤拷锟芥储锟斤拷锟矫★拷 -> 锟斤拷锟斤拷锟斤拷锟斤拷锟捷存储锟斤拷
锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟矫ｏ拷

锟斤拷锟截癸拷锟斤拷目录 锟斤拷锟皆讹拷锟藉创锟斤拷锟斤拷锟截癸拷锟斤拷目录锟斤拷锟斤拷锟斤拷使锟斤拷 /data/input
锟芥储位锟斤拷 锟斤拷选锟斤拷锟斤拷锟斤拷锟斤拷锟侥诧拷锟斤拷锟侥硷拷系统锟斤拷锟斤拷锟斤拷选锟斤拷锟窖达拷锟斤拷锟斤拷 moderarts0009 锟斤拷锟斤拷锟皆硷拷锟斤拷锟捷硷拷锟斤拷锟节碉拷目录 input

锟斤拷锟斤拷Notebook锟斤拷锟斤拷 -> 锟斤拷锟斤拷锟斤拷蚩锟絥otebook-test ->
锟津开★拷 Terminal 锟斤拷锟斤拷锟斤拷锟斤拷锟秸端斤拷锟斤拷 ->
执锟斤拷锟斤拷锟斤拷锟斤拷锟筋，锟斤拷锟斤拷锟斤拷锟节诧拷锟皆碉拷test锟侥硷拷
touch /data/input/test
锟斤拷执锟斤拷锟斤拷锟斤拷锟斤拷锟筋，锟斤拷锟皆匡拷锟斤拷锟斤拷沾锟斤拷锟斤拷锟絫est锟侥硷拷&锟斤拷前锟较达拷锟斤拷锟侥硷拷
ls /data/input
锟较达拷
锟斤拷锟斤拷选锟斤拷 OBS锟侥硷拷锟较达拷 锟斤拷
锟斤拷为锟斤拷锟斤拷 锟斤拷锟斤拷锟较达拷 锟斤拷锟斤拷为100M锟侥硷拷锟斤拷
锟斤拷压
锟津开★拷 Terminal 锟斤拷锟斤拷锟斤拷锟斤拷锟秸端斤拷锟斤拷 ->
执锟斤拷锟斤拷锟斤拷锟斤拷锟筋，锟介看锟角凤拷锟斤拷锟斤拷确锟斤拷路锟斤拷锟斤拷
pwd
ls -l
执锟斤拷锟斤拷锟斤拷锟斤拷锟筋，锟斤拷压锟斤拷目锟斤拷锟斤拷锟侥硷拷压锟斤拷锟斤拷
锟斤拷锟斤拷锟斤拷锟皆癸拷业锟绞硷拷Unet为锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟缴参匡拷锟斤拷末 学习锟斤拷源锟狡硷拷 锟斤拷
unzip unet.zip
unzip unet_sdk.zip

训锟斤拷锟阶讹拷 锟斤拷锟斤拷 目录锟结构 锟斤拷锟斤拷:

锟斤拷锟斤拷锟斤拷unet
	锟斤拷锟斤拷锟斤拷data                            // 预锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷菁锟斤拷募锟斤拷锟�
	锟斤拷锟斤拷锟斤拷raw_data                        // 原始锟斤拷锟捷硷拷
    锟斤拷锟斤拷锟斤拷out_model                       // 模锟酵碉拷锟斤拷锟斤拷锟斤拷锟侥硷拷锟斤拷
    锟斤拷锟斤拷锟斤拷pred_visualization              // 锟斤拷锟接伙拷图片锟斤拷锟斤拷锟侥硷拷锟叫ｏ拷锟斤拷要锟皆硷拷锟斤拷锟斤拷锟斤拷
    锟斤拷锟斤拷锟斤拷src                             // 锟斤拷锟杰猴拷锟斤拷
    锟斤拷   锟斤拷锟斤拷锟斤拷unet_medical                   // U-Net锟斤拷锟斤拷
    锟斤拷   锟斤拷锟斤拷锟斤拷unet_nested                    // U-Net++锟斤拷锟斤拷
    锟斤拷   锟斤拷锟斤拷锟斤拷config.py                      // 锟斤拷锟斤拷锟侥硷拷
    锟斤拷   锟斤拷锟斤拷锟斤拷data_loader.py                 // 锟斤拷锟捷硷拷锟斤拷
    锟斤拷   锟斤拷锟斤拷锟斤拷eval_callback.py               // 训锟斤拷时锟斤拷锟斤拷锟截碉拷
    锟斤拷   锟斤拷锟斤拷锟斤拷loss.py                        // 锟斤拷失锟斤拷锟斤拷
    锟斤拷   锟斤拷锟斤拷锟斤拷utils.py                       // 锟斤拷锟斤拷锟洁函锟斤拷
    锟斤拷锟斤拷锟斤拷draw_result_folder.py           // 锟侥硷拷锟斤拷图片锟斤拷锟接伙拷
    锟斤拷锟斤拷锟斤拷draw_result_single.py           // 锟斤拷锟斤拷图片锟斤拷锟接伙拷
    锟斤拷锟斤拷锟斤拷eval.py                         // 模锟斤拷锟斤拷证
    锟斤拷锟斤拷锟斤拷export.py                       // 模锟酵碉拷锟斤拷锟斤拷ckpt转air/mindir/onnx
    锟斤拷锟斤拷锟斤拷postprocess.py                  // 锟斤拷锟斤拷
    锟斤拷锟斤拷锟斤拷preprocess.py                   // 前锟斤拷锟斤拷
    锟斤拷锟斤拷锟斤拷preprocess_dataset.py           // 锟斤拷锟捷硷拷预锟斤拷锟斤拷
    锟斤拷锟斤拷锟斤拷train.py                        // 模锟斤拷训锟斤拷
    锟斤拷锟斤拷锟斤拷requirements.txt

模锟斤拷转锟斤拷 锟斤拷锟斤拷 目录锟结构 锟斤拷锟斤拷:

锟斤拷锟斤拷锟斤拷 unet_sdk
    锟斤拷锟斤拷锟斤拷 model
    锟斤拷   锟斤拷锟斤拷锟斤拷air2om.sh                     // air模锟斤拷转om锟脚憋拷
    锟斤拷   锟斤拷锟斤拷锟斤拷xxx.air                       //训锟斤拷锟阶段碉拷锟斤拷锟斤拷air模锟斤拷
    锟斤拷   锟斤拷锟斤拷锟斤拷aipp_unet_simple_opencv.cfg   // aipp锟侥硷拷

注锟斤拷
锟斤拷锟斤拷锟斤拷锟酵匡拷锟皆匡拷始锟矫程ｏ拷锟斤拷锟斤拷训锟斤拷锟阶段★拷

锟斤拷锟斤拷途锟斤拷停实锟介，锟角碉拷锟斤拷停止锟斤拷源锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟劫凤拷锟矫ｏ拷
锟斤拷锟斤拷锟截硷拷锟斤拷实锟介，锟劫达拷锟斤拷锟斤拷Notebook锟斤拷袒锟斤拷锟斤拷锟�
锟斤拷锟斤拷锟斤拷吮锟绞碉拷椋拷锟斤拷锟斤拷锟酵凤拷锟斤拷源锟斤拷锟斤拷锟斤拷为锟斤拷停止锟狡费★拷

一. 锟斤拷锟斤拷锟侥硷拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷预锟斤拷锟斤拷

MindSpore 锟斤拷锟捷硷拷预锟斤拷锟斤拷preprocess_dataset.py锟侥硷拷锟斤拷锟斤拷锟斤拷锟斤拷陆疟锟�:

锟侥硷拷锟斤拷锟斤拷锟脚憋拷src/config.py锟侥硷拷锟斤拷

锟侥硷拷锟斤拷锟斤拷锟脚憋拷为src/config.py锟斤拷锟斤拷锟斤拷
unet_medical锟斤拷
unet_nested锟斤拷
unet_nested_cell锟斤拷
unet_simple锟斤拷
unet_simple_coco
锟斤拷5锟斤拷锟斤拷锟矫ｏ拷锟斤拷示模锟斤拷锟斤拷锟斤拷锟捷硷拷之锟斤拷锟斤拷锟较★拷
锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟捷硷拷路锟斤拷锟斤拷锟侥硷拷锟斤拷锟斤拷

Step 锟斤拷锟叫脚憋拷

锟铰斤拷NoteBook锟叫ｏ拷锟介看锟角凤拷锟节癸拷锟斤拷目录unet/路锟斤拷锟斤拷
!pwd
锟斤拷锟斤拷NoteBook锟叫ｏ拷锟斤拷锟斤拷示锟斤拷
!python3 preprocess_dataset.py --data_url=./data/
锟斤拷锟斤拷--data_url锟斤拷锟斤拷锟捷硷拷预锟斤拷锟斤拷锟斤拷谋锟斤拷锟铰凤拷锟斤拷锟�

预锟斤拷锟斤拷锟捷硷拷预锟斤拷锟斤拷锟斤拷锟斤拷时锟斤拷约为10锟斤拷锟接★拷
预锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷菁锟斤拷岜ｏ拷锟斤拷锟�/unet/data/锟侥硷拷锟斤拷锟铰★拷
锟斤拷锟斤拷锟斤拷锟斤拷

锟斤拷. 模锟斤拷训锟斤拷

MindSpore模锟斤拷训锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷陆疟锟�:

preprocess_dataset.py:锟斤拷锟斤拷coco锟斤拷锟捷硷拷 转锟斤拷锟斤拷 模锟斤拷训锟斤拷锟斤拷要锟斤拷锟捷革拷式锟斤拷
src/unet_xxx/:锟斤拷锟� unet/unet++ 模锟酵结构锟斤拷
src/data_loader.py:锟斤拷锟� 锟斤拷锟捷硷拷锟截癸拷锟杰猴拷锟斤拷锟斤拷
src/eval_callback:锟斤拷锟� cb 锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷训锟斤拷锟斤拷锟斤拷锟叫斤拷锟斤拷eval.
src/utils.py: mindspore 锟皆讹拷锟斤拷 cb 锟斤拷锟斤拷锟斤拷锟皆讹拷锟斤拷 metrics 锟斤拷锟斤拷锟斤拷
train.py

Step 锟斤拷锟叫脚憋拷

锟斤拷锟斤拷NoteBook锟叫ｏ拷锟斤拷锟斤拷示锟斤拷
!python train.py --data_url=./data/ --run_eval=True
锟斤拷锟斤拷--data_url锟斤拷锟斤拷锟捷硷拷锟斤拷锟斤拷路锟斤拷锟斤拷
锟斤拷锟斤拷--run_eval: True 锟斤拷示训锟斤拷锟斤拷锟斤拷锟斤拷同时锟斤拷锟斤拷锟斤拷证锟斤拷

预锟斤拷模锟斤拷训锟斤拷锟斤拷锟斤拷时锟斤拷约为36锟斤拷锟接★拷
锟斤拷锟斤拷锟斤拷锟斤拷

锟斤拷. 模锟斤拷锟斤拷锟斤拷

MindSpore模锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷陆疟锟�:

src/unet_xxx/锟斤拷锟斤拷锟絬net/unet++模锟酵结构锟斤拷
src/data_loader.py锟斤拷锟斤拷锟斤拷锟斤拷锟皆わ拷锟斤拷锟斤拷锟斤拷锟斤拷菁锟斤拷毓锟斤拷芎锟斤拷锟斤拷锟�
src/utils.py锟斤拷mindspore锟皆讹拷锟斤拷cb锟斤拷锟斤拷锟斤拷锟皆讹拷锟斤拷metrics锟斤拷锟斤拷锟斤拷
eval.py

Step 锟斤拷锟叫脚憋拷

锟斤拷锟斤拷NoteBook锟叫ｏ拷锟斤拷锟斤拷示锟斤拷
!python eval.py --data_url=./data/ --ckpt_path=./ckpt_0/best.ckpt
锟斤拷锟斤拷--data_url锟斤拷锟斤拷锟捷硷拷锟斤拷锟斤拷路锟斤拷锟斤拷
锟斤拷锟斤拷--ckpt_path锟斤拷ckpt 锟斤拷取路锟斤拷

预锟斤拷模锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷时锟斤拷约为2锟斤拷锟接★拷
锟斤拷锟斤拷锟斤拷锟斤拷

注锟斤拷
IOU锟斤拷Intersection over Union锟斤拷锟斤拷一锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷
锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷呓锟斤拷锟斤拷氐锟斤拷潭龋锟饺≈碉拷锟轿猍0,1]锟斤拷锟斤拷
锟截碉拷锟斤拷锟斤拷锟斤拷越锟斤拷IOU值锟斤拷越锟斤拷

锟斤拷. 锟斤拷锟斤拷锟斤拷踊锟�

锟斤拷锟斤拷通锟斤拷锟斤拷图锟侥凤拷式锟斤拷图锟斤拷慕锟斤拷锟斤拷锟接伙拷锟斤拷锟斤拷锟斤拷榭达拷锟�
锟斤拷锟接伙拷锟斤拷锟斤拷锟斤拷锟斤拷锟街★拷

锟斤拷锟斤拷一锟斤拷锟斤拷图片锟斤拷锟接伙拷

draw_result_single.py锟斤拷 锟斤拷锟斤拷图片锟斤拷锟接伙拷 锟斤拷
锟斤拷锟斤拷锟斤拷锟酵计拷牟眉锟斤拷锟酵硷拷锟斤拷crop_plot.png锟斤拷模锟斤拷预锟斤拷慕锟斤拷predict_plot.png锟斤拷

Step 锟斤拷锟叫脚憋拷

锟介看锟斤拷锟斤拷目录unet/路锟斤拷锟斤拷
确锟斤拷锟窖撅拷锟斤拷锟饺达拷锟斤拷锟斤拷
锟斤拷锟接伙拷图片锟斤拷锟斤拷锟侥硷拷 pred_visualization锟侥硷拷锟斤拷
锟斤拷锟斤拷NoteBook锟叫ｏ拷锟斤拷锟斤拷示锟斤拷
!python draw_result_single.py --data_url=./data/SW1hZ2VfMjAyMTA3MjcxNTEzMzYzNzk --save_url=./pred_visualization --ckpt_path=./ckpt_0/best.ckpt
锟斤拷锟斤拷--data_url锟斤拷锟斤拷锟捷硷拷锟斤拷锟斤拷路锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷图锟今）★拷
锟斤拷锟斤拷--save_url锟斤拷锟斤拷锟酵硷拷癖４锟铰凤拷锟斤拷锟�
锟斤拷锟斤拷--ckpt_path锟斤拷ckpt锟斤拷取路锟斤拷锟斤拷

锟斤拷锟斤拷图片锟斤拷锟接伙拷锟斤拷锟斤拷时锟斤拷约为1锟斤拷锟接★拷
锟斤拷锟接伙拷锟斤拷锟酵计拷岜ｏ拷锟斤拷锟�/unet/pred_visualization锟侥硷拷锟斤拷锟铰★拷
锟斤拷锟斤拷锟斤拷锟斤拷

锟斤拷锟斤拷锟斤拷锟侥硷拷锟斤拷图片锟斤拷锟接伙拷

draw_result_folder.py锟斤拷 锟侥硷拷锟斤拷图片锟斤拷锟接伙拷 锟斤拷
锟斤拷锟斤拷募锟斤拷锟斤拷锟酵计拷锟侥ｏ拷锟皆わ拷锟斤拷锟絧redict.png锟斤拷

Step 锟斤拷锟叫脚憋拷

锟介看锟斤拷锟斤拷目录unet/路锟斤拷锟斤拷
确锟斤拷锟窖撅拷锟斤拷锟饺达拷锟斤拷锟斤拷
锟斤拷锟接伙拷图片锟斤拷锟斤拷锟侥硷拷 pred_visualization锟侥硷拷锟斤拷
锟斤拷锟斤拷NoteBook锟叫ｏ拷锟斤拷锟斤拷示锟斤拷
!python draw_result_folder.py --data_url=./data/ --save_url=./pred_visualization --ckpt_path=./ckpt_0/best.ckpt
锟斤拷锟斤拷--data_url锟斤拷锟斤拷锟捷硷拷锟斤拷锟斤拷路锟斤拷锟斤拷锟斤拷图锟斤拷锟侥硷拷锟叫ｏ拷锟斤拷
锟斤拷锟斤拷--save_url锟斤拷锟斤拷锟酵硷拷癖４锟铰凤拷锟斤拷锟�
锟斤拷锟斤拷--ckpt_path锟斤拷ckpt锟斤拷取路锟斤拷锟斤拷

锟侥硷拷锟斤拷图片锟斤拷锟接伙拷锟斤拷锟斤拷时锟斤拷约为10锟斤拷锟接★拷
锟斤拷锟接伙拷锟斤拷锟酵计拷岜ｏ拷锟斤拷锟�/unet/pred_visualization锟侥硷拷锟斤拷锟铰★拷
锟斤拷锟斤拷锟斤拷锟斤拷

锟斤拷. 模锟酵憋拷锟斤拷

锟斤拷锟斤拷锟斤拷跁N锟斤拷AI锟斤拷锟斤拷锟斤拷锟斤拷执锟斤拷锟斤拷锟斤拷锟斤拷
锟斤拷锟斤拷通锟斤拷锟斤拷锟界定锟斤拷锟紺heckPoint锟斤拷锟斤拷AIR锟斤拷式模锟斤拷锟侥硷拷锟斤拷

Step 锟斤拷锟叫脚憋拷

锟斤拷锟斤拷NoteBook锟叫ｏ拷锟斤拷锟斤拷示锟斤拷
!python export.py --ckpt_file="./ckpt_0/best.ckpt" --width=960 --height=960 --file_name="out_model/unet_hw960_bs1" --file_format="AIR"
锟斤拷锟叫–-ckpt_file锟斤拷 ckpt路锟斤拷锟斤拷
锟斤拷锟斤拷--width锟斤拷模锟斤拷锟斤拷锟斤拷叽纭�
锟斤拷锟斤拷--height锟斤拷模锟斤拷锟斤拷锟斤拷叽纭�
锟斤拷锟斤拷--file_name锟斤拷锟斤拷锟斤拷募锟斤拷锟斤拷锟�
锟斤拷锟斤拷--file_format锟斤拷锟斤拷锟斤拷锟绞斤拷锟斤拷锟斤拷锟轿猍锟斤拷ONNX锟斤拷, 锟斤拷AIR锟斤拷, 锟斤拷MINDIR锟斤拷]锟斤拷

模锟酵憋拷锟芥即锟斤拷锟斤拷模锟酵碉拷锟斤拷锟斤拷锟斤拷锟斤拷out_model/unet_hw960_bs1.air
锟斤拷蠼锟斤拷锟斤拷锟侥ｏ拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷兀锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锥锟绞碉拷锟绞癸拷茫锟�
锟揭硷拷 -> Download

锟斤拷. 模锟斤拷转锟斤拷

锟剿达拷模锟斤拷转锟斤拷锟斤拷要锟矫碉拷ATC锟斤拷锟竭★拷
锟斤拷细锟斤拷锟斤拷&锟斤拷锟斤拷锟斤拷锟斤拷慰锟� 锟絅锟节癸拷锟斤拷锟侥碉拷-使锟斤拷ATC锟斤拷锟斤拷转锟斤拷模锟斤拷

Step1 锟较达拷air模锟斤拷

锟斤拷训锟斤拷锟阶讹拷实锟斤拷模锟酵憋拷锟斤拷锟� air模锟斤拷 锟较达拷锟斤拷锟斤拷为锟斤拷ModelArts锟斤拷unet_sdk/model/目录锟斤拷

锟斤拷锟斤拷锟斤拷为模锟斤拷锟斤拷锟斤拷optype[ArgMaxD]锟斤拷锟斤拷锟斤拷锟揭拷锟紸scend910系锟斤拷芯片锟斤拷执锟斤拷模锟斤拷转锟斤拷锟斤拷锟杰成癸拷锟斤拷
锟斤拷锟剿次伙拷为锟斤拷ModelArts使锟矫碉拷锟斤拷锟斤拷Ascend910A锟斤拷
锟斤拷一锟斤拷锟斤拷锟斤拷锟侥ｏ拷锟窖碉拷锟斤拷锟斤拷锟叫碉拷模锟斤拷转锟斤拷锟角匡拷锟斤拷选锟斤拷锟节匡拷锟斤拷锟斤拷锟阶硷拷(Ascend310系锟斤拷芯片)锟斤拷Ubuntu系统锟斤拷执锟叫的★拷
锟斤拷锟斤拷锟藉方锟斤拷锟斤拷慰锟� 锟絅锟节癸拷锟斤拷锟侥碉拷-转锟斤拷模锟斤拷锟斤拷

Step2 模锟斤拷转锟斤拷锟斤拷锟斤拷

锟斤拷unet_sdk/model/air2om.sh锟侥硷拷
使锟斤拷atc锟斤拷锟斤拷锟斤拷锟铰ｏ拷锟缴革拷锟斤拷实锟绞匡拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷薷摹锟�

atc --framework=1 --model=unet_hw960_bs1.air --output=unet_hw960_bs1 --input_format=NCHW --soc_version=Ascend910A --log=error --insert_op_conf=aipp_unet_simple_opencv.cfg

锟斤拷实锟介将 训锟斤拷锟阶讹拷 实锟斤拷 模锟酵憋拷锟斤拷 锟斤拷 air模锟斤拷 转为锟絅锟斤拷Al锟斤拷锟斤拷锟斤拷支锟街碉拷 om锟斤拷式锟斤拷锟斤拷模锟斤拷
注锟解：air 模锟斤拷转 om 只支锟街撅拷态 batch锟斤拷锟斤拷锟斤拷 batchsize=1锟斤拷
锟斤拷锟斤拷--framework锟斤拷原始锟斤拷锟斤拷锟斤拷汀锟�
锟斤拷锟斤拷--model锟斤拷原始模锟斤拷锟侥硷拷路锟斤拷锟斤拷锟侥硷拷锟斤拷锟斤拷
锟斤拷锟斤拷--output锟斤拷转锟斤拷锟斤拷锟斤拷锟斤拷锟侥ｏ拷偷锟铰凤拷锟斤拷约锟斤拷募锟斤拷锟斤拷锟�
锟斤拷锟斤拷--input_format锟斤拷锟斤拷锟斤拷锟斤拷锟捷革拷式锟斤拷
锟斤拷锟斤拷--soc_version锟斤拷模锟斤拷转锟斤拷时指锟斤拷芯片锟芥本锟斤拷
锟斤拷 锟斤拷浠爸革拷锟斤拷堑锟角爸达拷锟侥ｏ拷锟阶拷锟绞憋拷锟斤拷锟斤拷诨锟斤拷锟斤拷锟叫酒拷姹� 锟斤拷锟斤拷通锟斤拷锟斤拷锟斤拷锟斤拷锟秸讹拷锟斤拷锟斤拷 npu-smi info 锟介看锟斤拷
锟斤拷锟斤拷--log锟斤拷锟斤拷示锟斤拷志锟侥硷拷锟斤拷
锟斤拷锟斤拷--insert_op_conf锟斤拷锟斤拷锟斤拷锟斤拷锟接碉拷锟斤拷锟斤拷锟侥硷拷路锟斤拷锟斤拷锟侥硷拷锟斤拷锟斤拷锟斤拷锟斤拷使锟斤拷AIPP预锟斤拷锟斤拷锟斤拷锟斤拷锟侥硷拷锟斤拷锟斤拷锟斤拷图锟斤拷锟斤拷锟斤拷预锟斤拷锟斤拷锟斤拷

Step3 锟斤拷锟叫脚憋拷

确锟斤拷锟节癸拷锟斤拷目录unet_sdk/model/路锟斤拷锟铰ｏ拷锟斤拷锟饺查看锟侥硷拷权锟斤拷
ls -l
锟斤拷锟斤拷锟斤拷募锟饺拷锟斤拷锟斤拷锟矫伙拷锟絰锟斤拷锟斤拷锟斤拷锟揭拷锟斤拷锟斤拷锟揭伙拷锟斤拷罡筹拷锟斤拷锟街达拷锟饺拷蓿锟�
锟斤拷锟斤拷
chmod +x air2om.sh
锟斤拷锟斤拷示锟斤拷
锟斤拷锟斤拷
./air2om.sh

锟斤拷锟斤拷锟斤拷锟斤拷

注锟斤拷
锟斤拷锟斤拷锟斤拷锟斤拷锟节伙拷为锟斤拷锟斤拷使锟斤拷MindSpore锟斤拷训锟斤拷锟阶讹拷实锟斤拷徒锟斤拷锟斤拷恕锟�
锟斤拷锟剿碉拷锟斤拷锟斤拷air模锟酵硷拷锟斤拷模锟斤拷转锟斤拷锟斤拷锟斤拷om模锟酵ｏ拷锟斤拷锟角就匡拷锟皆硷拷锟斤拷锟斤拷锟斤拷锟斤拷一篇锟铰ｏ拷 AscendCL锟斤拷锟斤拷锟阶讹拷 锟斤拷

锟斤拷锟斤拷锟斤拷堑锟� 锟斤拷时锟截憋拷锟斤拷锟较伙拷锟斤拷 锟斤拷锟斤拷锟斤拷锟斤拷源锟剿费和诧拷锟斤拷锟斤拷锟斤拷姆锟斤拷茫锟斤拷锟斤拷锟�

学习锟斤拷源锟狡硷拷

锟絅思锟斤拷锟斤拷锟教筹拷1.9锟斤拷模锟斤拷训锟斤拷
GitHub锟斤拷mindspore-ai / models

小锟斤拷锟狡硷拷锟侥讹拷

锟斤拷页

锟斤拷锟斤拷戏

锟斤拷戏锟斤拷

锟斤拷锟斤拷锟�

锟斤拷锟斤拷锟�

锟斤拷锟斤拷锟斤拷

锟斤拷锟斤拷锟斤拷锟叫帮拷

锟斤拷锟斤拷锟斤拷锟斤拷

锟斤拷戏锟斤拷锟斤拷

锟斤拷锟斤拷专锟斤拷

锟斤拷锟斤拷频锟斤拷

小锟斤拷锟狡硷拷

锟斤拷色锟斤拷目

锟斤拷锟斤拷锟斤拷锟斤拷

一锟斤拷999

锟斤拷锟截吃硷拷

沙锟斤拷锟斤拷戏

BT锟斤拷锟斤拷

锟斤拷锟斤拷只锟�

锟斤拷锟剿伙拷锟斤拷

锟絅锟节匡拷锟斤拷全锟斤拷锟斤拷 之 MindSpore锟斤拷为锟斤拷模锟斤拷训锟斤拷

前锟斤拷

训锟斤拷锟阶讹拷

A. 锟斤拷锟斤拷锟筋建

Step1 锟斤拷锟斤拷OBS锟斤拷锟斤拷锟侥硷拷

Step2 锟较达拷锟斤拷锟斤拷锟侥硷拷锟斤拷OBS锟斤拷锟斤拷锟侥硷拷系统

Step3 锟斤拷锟斤拷ModelArts锟斤拷锟斤拷Notebook锟斤拷袒锟斤拷锟�

Step4 为Notebook锟斤拷袒锟斤拷锟斤拷锟斤拷锟窖碉拷锟斤拷锥锟斤拷锟侥匡拷锟斤拷锟斤拷募锟�

一. 锟斤拷锟斤拷锟侥硷拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷预锟斤拷锟斤拷

Step 锟斤拷锟叫脚憋拷

锟斤拷. 模锟斤拷训锟斤拷

Step 锟斤拷锟叫脚憋拷

锟斤拷. 模锟斤拷锟斤拷锟斤拷

Step 锟斤拷锟叫脚憋拷

锟斤拷. 锟斤拷锟斤拷锟斤拷踊锟�

锟斤拷锟斤拷一 锟斤拷锟斤拷图片锟斤拷锟接伙拷

Step 锟斤拷锟叫脚憋拷

锟斤拷锟斤拷锟斤拷 锟侥硷拷锟斤拷图片锟斤拷锟接伙拷

Step 锟斤拷锟叫脚憋拷

锟斤拷. 模锟酵憋拷锟斤拷

Step 锟斤拷锟叫脚憋拷

锟斤拷. 模锟斤拷转锟斤拷

Step1 锟较达拷air模锟斤拷

Step2 模锟斤拷转锟斤拷锟斤拷锟斤拷

Step3 锟斤拷锟叫脚憋拷

学习锟斤拷源锟狡硷拷

锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟侥斤拷为锟斤拷锟斤拷锟斤拷息锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷同锟斤拷锟睫观碉拷锟街な碉拷锟斤拷锟斤拷锟斤拷锟�

锟斤拷锟斤拷锟狡碉拷锟斤拷锟�

锟斤拷锟斤拷

同锟斤拷锟斤拷锟斤拷

锟斤拷锟斤拷

锟斤拷锟斤拷锟斤拷讯

锟斤拷锟斤拷

锟斤拷锟斤拷

锟斤拷锟斤拷

锟絅锟节匡拷锟斤拷全锟斤拷锟斤拷之 MindSpore锟斤拷为锟斤拷模锟斤拷训锟斤拷

锟斤拷锟斤拷一锟斤拷锟斤拷图片锟斤拷锟接伙拷

锟斤拷锟斤拷锟斤拷锟侥硷拷锟斤拷图片锟斤拷锟接伙拷