|  |  |  | @ -1,4 +1,4 @@ | 
			
		
	
		
			
				
					|  |  |  |  | # Exports a pytorch *.pt model to *.onnx format. Example usage: | 
			
		
	
		
			
				
					|  |  |  |  | # Exports a pytorch *.pt model to *.onnx format. Example usage (run from ./yolov5 directory): | 
			
		
	
		
			
				
					|  |  |  |  | # $ export PYTHONPATH="$PWD" | 
			
		
	
		
			
				
					|  |  |  |  | # $ python models/onnx_export.py --weights ./weights/yolov5s.pt --img 640 --batch 1 | 
			
		
	
		
			
				
					|  |  |  |  | 
 | 
			
		
	
	
		
			
				
					|  |  |  | @ -10,7 +10,7 @@ from models.common import * | 
			
		
	
		
			
				
					|  |  |  |  | 
 | 
			
		
	
		
			
				
					|  |  |  |  | if __name__ == '__main__': | 
			
		
	
		
			
				
					|  |  |  |  |     parser = argparse.ArgumentParser() | 
			
		
	
		
			
				
					|  |  |  |  |     parser.add_argument('--weights', default='./weights/yolov5s.pt', help='model path RELATIVE to ./models/') | 
			
		
	
		
			
				
					|  |  |  |  |     parser.add_argument('--weights', default='./weights/yolov5s.pt', help='weights path') | 
			
		
	
		
			
				
					|  |  |  |  |     parser.add_argument('--img-size', default=640, help='inference size (pixels)') | 
			
		
	
		
			
				
					|  |  |  |  |     parser.add_argument('--batch-size', default=1, help='batch size') | 
			
		
	
		
			
				
					|  |  |  |  |     opt = parser.parse_args() | 
			
		
	
	
		
			
				
					|  |  |  | 
 |