If not specified, the results will not be saved to a file.ĮVAL_METRICS: Items to be evaluated on the results. RESULT_FILE: Filename of the output results in pickle format. Tools/dist_test.sh also supports multi-node testing, but relies on PyTorch’s launch utility. show_result ( img, result, out_file = 'result.jpg' ) asyncio. show_result ( img, result ) # or save the visualization results to image files model. Stream ( device = device )) # test a single image and show the results img = 'test.jpg' # or img = mmcv.imread(img), which will only load it once async with concurrent ( streamqueue ): result = await async_inference_detector ( model, img ) # visualize the results in a new window model. Queue () # queue size defines concurrency level streamqueue_size = 3 for _ in range ( streamqueue_size ): streamqueue. Import asyncio import torch from mmdet.apis import init_detector, async_inference_detector from import concurrent async def main (): config_file = 'configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py' checkpoint_file = 'checkpoints/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth' device = 'cuda:0' model = init_detector ( config_file, checkpoint = checkpoint_file, device = device ) # queue is used for concurrent inference of multiple images streamqueue = asyncio. Note: inference_detector only supports single-image inference for now. show_result ( frame, result, wait_time = 1 )Ī notebook demo can be found in demo/inference_demo.ipynb. VideoReader ( 'video.mp4' ) for frame in video : result = inference_detector ( model, frame ) model.

show_result ( img, result, out_file = 'result.jpg' ) # test a video and show the results video = mmcv. Tutorial 9: ONNX to TensorRT (Experimental)įrom mmdet.apis import init_detector, inference_detector import mmcv # Specify the path to model config and checkpoint file config_file = 'configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py' checkpoint_file = 'checkpoints/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth' # build the model from a config file and a checkpoint file model = init_detector ( config_file, checkpoint_file, device = 'cuda:0' ) # test a single image and show the results img = 'test.jpg' # or img = mmcv.imread(img), which will only load it once result = inference_detector ( model, img ) # visualize the results in a new window model.

Tutorial 8: Pytorch to ONNX (Experimental).3: Train with customized models and standard datasets.1: Inference and train with existing models and standard datasets.

Run 8 v2 monolith map