[MYAI Studio SDK] Image-Pix2Pix-PyTorch-Jupyter

Pix2Pix style conversion. Output the original image to the learned image style. For example: convert black and white to color, edge map to photo, map to satellite map and other applications. 

[Instructions]

1_combine_A_and_B.ipynb

There are train, val, and test in the data/A and data/B folders. A is the image after conversion, and B is the image before conversion.

Please put the images into the train, val, and test in the A and B folders before and after the conversion. One thing to note is that the train, val, and test in the two folders must be able to find the corresponding image file names.

Then click 1. combine A and B to combine the corresponding images into one, and output to data/train, data/val, data/test. 

Note:

1). It is recommended that the length and width of the image be the same. If they are not the same, you can zoom or crop them.

2). The image file must be .jpg.

3). The images need to be matched in pairs. 

2_visdom_server.ipynb

Open visdom_server with port 8801. 

3_visdom_server_browser.ipynb

Open the browser, you can see the loss curve and other information of the training process. 

4_train.ipynb

Train the model. 

•    --continue_train : Load model/latest model to continue training. 

•    --n_epochs 1000 : The training epochs is 1000. 

•    --gpu_ids 0 : Which gpu to use for training, you can also set more than one, for example --gpu_ids 0,1,2.

•    --save_epoch_freq 5 : Store model frequency. 

• --epoch_count 2300: The output model file name starts from 2300 times, which means that there have been 2300 epochs of training before. It can be used with continue_train. 

5_inference_folder.ipynb

Inference folder. 

•    checkpoints_dir = 'model' : Model folder.

•    epoch ='latest': Load the latest_net_G_A model. 

•    dataroot ='data/B', phase ='test': The test data folder is data/B/test. 

•   num_test = 27: How many test images are in the folder. 

6_inference_folder_image_pairs.ipynb

Inferred folder, the images in the folder are the images of A and B combined by running 1_combine_A_and_B.ipynb.

•   checkpoints_dir ='model', epoch ='latest': The model is model/latest_net_G_A.pth.

•   dataroot ='data/', phase ='test': The test folder is data/test.

•   num_test = 20: How many test images are in the folder. 

inference result.png

This SDK is built in AppForAI - AI Dev Tools.

Purchase license separately: USD 600, permanent authorization, single APP authorization, single machine authorization, one-year activation, one-year download, one-year update, one-year email technical support.

Further Reading

1.
MYAI Studio for Windows

2.
MYAI Studio for Linux

3.
AI PC

Thanks for the support from our clients

National Taiwan University, National Tsing Hua University, National Yang Ming Chiao Tung University, National Cheng Kung University, Taipei Medical University, Kaohsiung Medical University, National Taipei University of Nursing and Health Sciences, China Medical University, National Chung Hsing University, National Central University, National Sun Yat-sen University, National Chung Cheng University, National Chi Nan University, National Chiayi University, National Ilan University, National Taipei University of Education, National United University, Tamkang University, Feng Chia University, Chang Gung University, I-Shou University, Shih Chien University, Tatung University, Chung Yuan Christian University, Soochow University, Tzu Chi University, Tzu Chi University of Science and Technology, National Taiwan University of Science and Technology, National Taipei University of Technology, National Taichung University of Science and Technology, National Yunlin University of Science and Technology, National Chin-Yi University of Technology, National Formosa University, National Pingtung University of Science and Technology, National Kaohsiung University of Science and Technology, Chaoyang University of Technology, Ming Chi University of Technology, Minghsin University of Science and Technology, Southern Taiwan University of Science and Technology, Asia Eastern University of Science and Technology, China University of Technology, National Taiwan Sport University, National Defense University, Republic of China Naval Academy, Republic of China Army Academy, Luodong Senior High School, Gushan Senior High School, Kaohsiung Girls Senior High School, National Taiwan University Hospital, National Cheng Kung University Hospital, Taipei Veterans General Hospital, Chang Gung Memorial Hospital, Tzu Chi Hospital, E-Da Hospital, Far Eastern Memorial Hospital, Lien Hsin International Hospital, National Chung-Shan Institute of Science and Technology, Armaments Bureau, Ministry of National Defense, Ministry of Justice Investigation Bureau, Industrial Technology Research Institute, Institute for Information Industry, Institute of Nuclear Energy Research, Atomic Energy Council, Endemic Species Research Institute, Council of Agriculture, Institute of Labor, Occupational Safety and Health, Ministry of Labor, Taiwan Textile Research Institute, Metal Industries Research & Development Centre, Taiwan Instrument Research Institute, Automotive Research & Testing Center, Chunghwa Telecom, Taiwan Water Corporation, Taiwan Semiconductor Manufacturing Company, United Microelectronics Corporation, Nanya Technology Corporation, Winbond Electronics Corporation, Wafer Works Corporation, Grandtop Optoelectronics, AU Optronics, Innolux Corporation, HannStar Display Corporation, Formosa Plastics Corporation, Formosa Petrochemical Corporation, Formosa Technologies Corporation, Nan Ya Plastics Corporation, Formosa Chemicals & Fibre Corporation, CPC Corporation, Taiwan, Logitech, Elan Microelectronics Corporation, Lextar Electronics Corporation, Darfon Electronics Corporation, Sercomm Corporation, EZconn Corporation, Foxconn Technology Group, WPG Holdings, World Peace Industrial Co., Ltd., Mirle Automation Corporation, Shuttle Machinery Co., Ltd., ChipMOS Technologies Inc., Simplo Technology Co., Ltd., Primax Electronics Ltd., Intesys Technology, Feng Hsin Iron & Steel Co., Ltd., China Ecotek Corporation, Zhu Sheng Technology, Advantech Co., Ltd., Stark Technology Inc., Hong Teng Technology, MiTAC International Corp., SYNNEX Technology International Corp., Allion Labs, Inc., Co-in International, Feng An Auto, Hong Hu International, Blue Ocean Intelligence, TOPPAN IDGATE Co., LTd., Far EasTone Telecommunications, SYSTEX Corporation, Cooler Master, Lion Travel, Beigang Wude Temple, Xiluo Fuxing Temple, etc.