Exemplo n.º 1
0
def smallnetv2_pool4_pose_crp192_rawImSz256(isRun=False,
                                            isGray=False,
                                            numTrain=1e+7,
                                            deviceId=[0],
                                            isPythonLayer=False,
                                            runNum=0):
    prms = sp.get_prms_pose(geoFence='dc-v2',
                            crpSz=192,
                            rawImSz=256,
                            splitDist=100,
                            numTrain=numTrain)
    nPrms = se.get_nw_prms(imSz=101,
                           netName='smallnet-v2',
                           concatLayer='pool4',
                           lossWeight=10.0,
                           randCrop=False,
                           concatDrop=False,
                           isGray=isGray,
                           isPythonLayer=isPythonLayer)
    lPrms = se.get_lr_prms(batchsize=256, stepsize=10000, clip_gradients=1.0)
    cPrms = se.get_caffe_prms(nPrms, lPrms, deviceId=deviceId, runNum=runNum)
    if isRun:
        exp = se.make_experiment(prms, cPrms)
        exp.run()
    else:
        return prms, cPrms
Exemplo n.º 2
0
def smallnetv5_pose_crp192_fc5_rawImSz256(isRun=False,
                                          isGray=False,
                                          numTrain=1e+7,
                                          deviceId=[0],
                                          isPythonLayer=True,
                                          runNum=0,
                                          extraFc=None,
                                          numFc5=None,
                                          lrAbove=None):
    prms = sp.get_prms_pose(geoFence='dc-v2',
                            crpSz=192,
                            rawImSz=256,
                            splitDist=100,
                            numTrain=numTrain)
    nPrms = se.get_nw_prms(imSz=101,
                           netName='smallnet-v5',
                           concatLayer='fc5',
                           lossWeight=10.0,
                           randCrop=False,
                           concatDrop=False,
                           isGray=isGray,
                           isPythonLayer=isPythonLayer,
                           numFc5=numFc5,
                           extraFc=extraFc,
                           lrAbove=lrAbove)
    lPrms = se.get_lr_prms(batchsize=256, stepsize=10000, clip_gradients=1.0)
    cPrms = se.get_caffe_prms(nPrms, lPrms, deviceId=deviceId, runNum=runNum)
    if isRun:
        exp = se.make_experiment(prms, cPrms)
        exp.run()
    return prms, cPrms
Exemplo n.º 3
0
def smallnet_pool4_pose(isRun=False):
	prms  = sp.get_prms_pose(geoFence='dc-v1')
	nPrms = se.get_nw_prms(imSz=101, netName='smallnet',
							 concatLayer='pool4')
	lPrms = se.get_lr_prms(batchsize=256)
	cPrms = se.get_caffe_prms(nPrms, lPrms, deviceId=[0,1])
	if isRun:
		exp   = se.make_experiment(prms, cPrms)
		exp.run()
	else:
		return prms, cPrms	
Exemplo n.º 4
0
def smallnetv3_pool4_pose_crp192(isRun=False):
	prms  = sp.get_prms_pose(geoFence='dc-v1', crpSz=192)
	nPrms = se.get_nw_prms(imSz=101, netName='smallnet-v3',
							 concatLayer='pool4', lossWeight=10.0)
	lPrms = se.get_lr_prms(batchsize=256, stepsize=10000, clip_gradients=10.0)
	cPrms = se.get_caffe_prms(nPrms, lPrms, deviceId=[0])
	if isRun:
		exp   = se.make_experiment(prms, cPrms)
		exp.run()
	else:
		return prms, cPrms	
Exemplo n.º 5
0
def smallnetv2_pool4_pose_randcrp(isRun=False):
	prms  = sp.get_prms_pose(geoFence='dc-v1')
	nPrms = se.get_nw_prms(imSz=101, netName='smallnet-v2',
							 concatLayer='pool4', randCrop=True)
	lPrms = se.get_lr_prms(batchsize=256, stepsize=5000)
	cPrms = se.get_caffe_prms(nPrms, lPrms, deviceId=[0,1,2,3])
	if isRun:
		exp   = se.make_experiment(prms, cPrms)
		exp.run()
	else:
		return prms, cPrms	
Exemplo n.º 6
0
def smallnetv2_pool4_pose_crp192_rawImSz256_newsplits(isRun=False, isGray=False):
	prms  = sp.get_prms_pose(geoFence='dc-v1', crpSz=192,
													 rawImSz=256, splitDist=100)
	nPrms = se.get_nw_prms(imSz=101, netName='smallnet-v2',
							 concatLayer='pool4', lossWeight=10.0,
								randCrop=False, concatDrop=False,
								isGray=isGray)
	lPrms = se.get_lr_prms(batchsize=256, stepsize=10000, clip_gradients=10.0)
	cPrms = se.get_caffe_prms(nPrms, lPrms, deviceId=[1])
	if isRun:
		exp   = se.make_experiment(prms, cPrms)
		exp.run()
	else:
		return prms, cPrms