foolishflyfox

Ignoring Files

# ignore files whose suffix is .o or .a
*.[oa]

# you can negate a pattern by starting it with an exclation point(!)
!hello.o

foolishflyfox

import torch
import torchvision
import torch.utils.data
from torch.backends import cudnn
from torchvision import transforms

# some hyperparameters setting
use_gpu = True
train_batch_size = 64
val_batch_size = 64

foolishflyfox

#!/usr/bin/env python

import argparse

parser = argparse.ArgumentParser(
    description='Some description about this script',
    formatter_class=argparse.ArgumentDefaultsHelpFormatter,  # show default value
)
# add required option. Note: required option can't with default parameter
parser.add_argument('info', type=str, help="required option")

foolishflyfox

Python

Deep Learning

Computer Vision

• train framework Detailed
• train framework simplified

foolishflyfox

import matplotlib.pyplot as plt
import numpy as np
import PIL

def show_images(images, rows = 1, titles = None):
    """Display a list of images in a single figure with matplotlib.
    
    Parameters
    ---------
    images: List of np.arrays compatible with plt.imshow.