當(dāng)前面已經(jīng)訓(xùn)練好修剪過的模型之后，接下來就可以在訓(xùn)練設(shè)備上先執(zhí)行推理計算，看看檢測的效果如何？如果實際推理結(jié)果并不理想的話，就得回到前面第 6 步驟重新再執(zhí)行一次。

9、執(zhí)行推理并查看結(jié)果

首先創(chuàng)建一個 test_samples 目錄，然后從數(shù)據(jù)集中 testing/image_2 中復(fù)制一些圖像過來測試，推理結(jié)果輸出到 ssd_infer_images 目錄下，這個推理計算也可以使用未修剪的模型來進(jìn)行測試。

在代碼框 “!tao ssd inference” 后面所提供的參數(shù)中，下面兩行是使用修剪后并重新訓(xùn)練的模型來進(jìn)行推理。

-e $SPECS_DIR/ssd_retrain_resnet18_kitti.txt  -m$USER_EXPERIMENT_DIR/experiment_dir_retrain/weights/ssd_resnet18_epoch_$EPOCH.tlt

如果將這兩行參數(shù)改成下面內(nèi)容，就是用原始模型來進(jìn)行推理。

-e $SPECS_DIR/ssd_train_resnet18_kitti.txt  -m$USER_EXPERIMENT_DIR/experiment_dir_unpruned/weights/ssd_resnet18_epoch_$EPOCH.tlt

這樣我們可以執(zhí)行透過推理結(jié)果，來查看二者的不同。下面兩張圖分別是未修剪（上）與修剪后（下）模型推理結(jié)果的局部放大，雖然修剪后的模型效果不如未修剪的，但是執(zhí)行結(jié)果還是令人滿意，當(dāng)然這得看您的要求有多高。

如果認(rèn)可這個推理結(jié)果的話，那么訓(xùn)練階段的任務(wù)就可以告一段落，接下去就準(zhǔn)備將模型導(dǎo)出，以便部署到真實的推理設(shè)備上。

10、導(dǎo)出模型

要執(zhí)行這個步驟，最好先確認(rèn)您的推理設(shè)備上所能支持的數(shù)據(jù)精度，特別是 INT8 類型，例如 Jetson Nano （含 2GB）、TX2/TX1就不支持，Xavier NX 與 AGX Xavier 是支持的。臺式機(jī)上的游戲卡一般不支持 INT8，而 Tesla、Quadro 專業(yè)卡是支持的，不過 NVIDIA 全系列產(chǎn)品都支持 FP16，如果不能確認(rèn)這方面的信息，至少選擇 FP16 是比較穩(wěn)妥。

簡單說明一下代碼框 “!tao ssd export” 后面的參數(shù)：

假如要輸出 INT8 模型時，由于需要特殊算法以及輔助的校準(zhǔn)工具，因此需要其他參數(shù)來協(xié)助這類模型的導(dǎo)出，主要如下：

這樣就能完成 INT8 模型的輸出任務(wù)。

到這里所輸出的 .etlt 模型文件基本上已經(jīng)可以部署到 DeepStream 推理設(shè)備上使用，因為 DeepStream 支持 .etlt 格式的調(diào)用，相當(dāng)方便。但如果要用其他獨立的 TensorRT 來執(zhí)行加速推理的話，就需要在推理設(shè)備上再進(jìn)行一次的轉(zhuǎn)換。

本步驟最后一個代碼塊 “!tao converter” 就是在訓(xùn)練設(shè)備上，將 .etlt 模型轉(zhuǎn)成這臺機(jī)器能使用的 TensorRT 引擎文件。由于參數(shù)內(nèi)容較多，這里不一一列舉，請自行至https://docs.nvidia.com/tao/tao-toolkit/text/object_detection/ssd.html搜索 “Using the tao-converter” 環(huán)節(jié)，有詳細(xì)的參數(shù)說明。

要注意的一點，在 “!tao converter” 所使用的“-t” 指定數(shù)據(jù)精度部分，需要與前面 “!tao ssd export” 的 “--data_type” 一致，否則會失去效果。

11、驗證部署模型的效果

這個步驟是驗證 TensorRT 加速引擎的推理效果，主要讓我們在部署到推理設(shè)備之前，能再確認(rèn)一下推理效果。

下面同樣提供兩張推理結(jié)果參照圖，上圖是未修剪的模型轉(zhuǎn)成 TensorRT 引擎的推理結(jié)果，下圖是修剪過的模型轉(zhuǎn)成 TensorRT 引擎的推理結(jié)果，效果看起來是一樣的，表示這個修剪過的模型還是足夠好的。

現(xiàn)在就完成一個物件檢測模型訓(xùn)練的過程，雖然看起來繁瑣，但是只要我們弄清楚一開始的路徑對應(yīng)，以及熟練后面每個環(huán)節(jié)的參數(shù)用法，基本上要比在框架中自行撰寫代碼容易得多了，除了收集與整理數(shù)據(jù)的工作是 TAO 工具無法協(xié)助的，其余部分已經(jīng)都為大家提供好完整的指令塊，這是一個真正 “Zero Coding” 的全方位模型訓(xùn)練工具。