鍥惧儚鍒嗙被妯″瀷璇勬祴妯″潡浣跨敤鏂规硶 ================================= 鍥惧儚鍒嗙被妯″瀷璇勬祴妯″潡鍙互閫氳繃璇诲彇 YAML 閰嶇疆鏂囦欢鐨勬柟寮忓湪鍛戒护琛屼腑浣跨敤锛屽懡浠ゅ涓嬶細 .. code-block:: bash python TestingImage.py /path/to/config.yml 閰嶇疆鏂囦欢鐨勭ず渚嬪彲瑙� \ `AI-Testing/image/configs/ExampleConfigFile.yml <https://openi.pcl.ac.cn/Numbda/AI-Testing/src/branch/main/image/configs/ExampleConfigFile.yml>`__\ 銆傚彲閰嶇疆鐨勪富瑕佸弬鏁板涓嬶細 .. code-block:: yaml # 璇勬祴绠楁硶锛屽彲浠ユ槸鍐呯疆鏂规硶涔嬩竴 # 涔熷彲浠ユ槸鐢ㄦ埛鑷繁瀹炵幇鐨勮瘎娴嬬畻娉� # 瀹炵幇鏂规硶鍙婅姹傝瑙佹枃妗� recipe: AutoPGD # 鎵板姩澶у皬锛屽崟浣嶄负鍍忕礌鍊� perturbation: 1 # 琚瘎娴嬫ā鍨嬬粨鏋�, 澶у皬鍐欐晱鎰� # 鍙互鏄唴缃殑妯″瀷 ResNet2, CIFAR10_RAND_enhanced 绛� # 鎴栬€呮槸鍐呯疆鎴栫敤鎴疯嚜瀹氫箟鐨勬ā鍨嬬粨鏋勶紝 # 渚嬪 Models.TestModel.resnet.ResNet18 # 涔熷彲浠ユ槸 torchvision 鏀寔鐨勬ā鍨� # 鍏蜂綋鏉ヨ鏄彲浠ヨ繑鍥炰竴涓ā鍨嬬殑鍑芥暟锛屼緥濡� # torchvision.models.resnet18 model: ResNet2 # 琚瘎娴嬫ā鍨嬫湰鍦拌矾寰� # 褰撴ā鍨嬬粨鏋勬槸鍐呯疆妯″瀷鏃讹紝妯″瀷璺緞鍙互鏄┖鍊� # 褰撴ā鍨嬬粨鏋勬槸 torchvision 鏀寔鐨勭粨鏋勬椂锛岃嫢璺緞涓虹┖鍊硷紝鍒欎粠 torchvision 涓嬭浇棰勮缁冩ā鍨� # 娉ㄦ剰锛� torchvision 妯″瀷榛樿鐨勯璁粌妯″瀷鏄湪ImageNet涓婅繘琛岀殑棰勮缁冿紝 # 姝ゆ椂 data_type 鍙傛暟鏈€濂借缃垚 ImageNet, # 鎴栬€呬紶鍏ヤ竴涓潪绌虹殑涓� data_type 閰嶅鐨� model_path model_path: null # 璇勬祴鐢ㄦ暟鎹紝鍐呯疆鏈� # cifar10_30, cifar10_100, cifar10_300, cifar10_1000, # ImageNetTiny锛�500寮犲浘锛�, ImageNetSmall锛�5000寮犲浘锛�, ImageCustom 绛� data_type: cifar10_1000 # 璇勬祴鐢ㄦ暟鎹湰鍦拌矾寰� # 鍙互鏄竴涓瓧鍏革紝涔熷彲浠ユ槸涓€涓垪琛紝渚嬪 # Data_path: # image_path: Datasets/CIFAR_cln_data/cifar10_30_origin_inputs.npy # label_path: Datasets/CIFAR_cln_data/cifar10_30_origin_labels.npy # 濡傛灉鏄唴缃暟鎹泦锛屽彲浠ユ槸绌哄€� Data_path: null # 璇勬祴鐢ㄦ暟鎹殑 class map 鏂囦欢璺緞 # 濡傛灉鏄唴缃暟鎹泦锛屽彲浠ユ槸绌哄€� Dict_path: null # 鏄惁鏄甫鐩爣鐨勫鎶楁敾鍑� IS_TARGETTED: false # 鏄惁鏄櫧鐩掑鎶楁敾鍑� IS_WHITE: true # 鏄惁淇濆瓨鐢熸垚鐨勫鎶楁牱鏈� IS_SAVE: false # 鐢熸垚鐨勫鎶楁牱鏈繚瀛樿矾寰勶紝鍙互鏄┖ save_path: null # 鎵瑰ぇ灏忥紝鍗冲苟琛屽鐞嗘牱鏈殑涓暟 batch_size: 32 # 鍒ゆ柇妯″瀷鏄惁椴佹鐨勬敾鍑绘垚鍔熺巼闃堝€� # 鍙栧€煎湪 (0, 1) 鍖洪棿鍐� robust_threshold: 0.6