分类预测笔记

  

   

    

     

    
    

     

      # -*- coding: utf-8 -*-
     
    
   

    

     

    
    

     

      from __future__ import print_function, division
     
    
   

    

     

    
    

     

      import argparse
     
    
   

    

     

    
    

     

      import os
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import cv2
     
    
   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import torch.nn.functional as m_func
     
    
   

    

     

    
    

     

      from torchvision import transforms
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import time
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      from demo.mobile_net import MobileNet2
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      parser = argparse.ArgumentParser(description='Training')
     
    
   

    

     

    
    

     

      parser.add_argument('--gpu_ids', default='0', type=str, help='gpu_ids: e.g. 0 0,1,2 0,2')
     
    
   

    

     

    
    

     

      parser.add_argument('--name', default='ft_ResNet50', type=str, help='output model name')
     
    
   

    

     

    
    

     

      parser.add_argument('--data_dir', default=r'\\192.168.25.73\Team-CV\dataset\2020_04_22_person', type=str, help='training dir path')
     
    
   

    

     

    
    

     

      parser.add_argument('--train_all', action='store_true', help='use all training data')
     
    
   

    

     

    
    

     

      parser.add_argument('--color_jitter', action='store_true', help='u
     
    
  
 

文章来源: blog.csdn.net，作者：AI视觉网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/105726960