checker.check_model(onnx_model) # check onnx model print('ONNX export success, saved as %s' % save_path) # Finish print('\nExport complete (%.2fs). Visualize with https://github.com/lutzroeder/netron.' % (time.time() - t)) def parse_opt(): parser = argparse.ArgumentParser() ...
stable-diffusion-webui/models/faceswaplab/parser/weights/facelib/parsing_parsenet.pthstable-diffusion-webui/models/faceswaplab/analysers/models/buffalo_lstable-diffusion-webui/models/faceswaplab/inswapper_128.onnx三个主要文件我都打包放到网盘了 有需要同学自取 网盘:pan.quark.cn/s/c73e9c14 ...
usage: onnx_inference.py [-h] [--onnx-weight ONNX_WEIGHT] [--input INPUT] [--output OUTPUT] Face parsing ONNX inference options: -h, --help show this help message and exit --onnx-weight ONNX_WEIGHT path to onnx model, default './weights/resnet18.onnx' --input INPUT path to...
@@ -113,7 +116,7 @@ def load_face_swapper_model(path="./assets/pretrained_models/inswapper_128.onnx" def load_face_parser_model(path="./assets/pretrained_models/79999_iter.pth"): global FACE_PARSER if FACE_PARSER is None: FACE_PARSER = init_parser(path, mode=device) FACE_PARSER ...
一是industry lab计算资源的统治。从Detectron的modelzoo里面所写:All baselines were run on Big Basin...
parser.add_argument('--noautoanchor', action='store_true', help='disable autoanchor check') parser.add_argument('--evolve', action='store_true', help='evolve hyperparameters') parser.add_argument('--bucket', type=str, default='', help='gsutil bucket') parser.add_argument('--cache-im...
parser.add_argument('--onnx_infer', action='store_true', default=True, help='onnx infer test') #===TensorRT=== parser.add_argument('--onnx2trt', action='store_true', default=True, help='export onnx to tensorrt') parser.add_argument('--fp16_trt', action='store_true', default...
parser.add_argument("--local_rank", type=int, default=-1, help="For distributed training: local_rank") parser.add_argument( "--checkpointing_steps", type=int, default=500, help=( "Save a checkpoint of the training state every X updates. These checkpoints are only...
UT_XMLSaxParser.h UT_XMLWriter.h UT_XNoise.h UT_XorShift.h UT_XXHash.h UT_ZString.h VCC VE VEX VGEO VIS VISF VM VOP VOPNET VRAY vulkan WIRE blosc-export.h blosc.h flicks.h fp_class.h glcorearb.h ieeefp.h png.h pngconf.h pnglibconf.h stdthread.h zconf.h File Members Examp...
Output : input mean and std: 127.5 127.5find model: ./insightface_func/models/antelope/glintr100.onnx recognitionfind model: ./insightface_func/models/antelope/scrfd_10g_bnkps.onnx detectionset det-size: (640, 640)0%| | 0/594 [00:00<?, ?it/s](142, 366, 4)100%|███████...