Python PCAM.get_D1_train示例

示例#1

文件： PCAMTrainALILikeVAEMSE.py 项目： caotians1/OD-test-master

from __future__ import print_function
import os,sys,inspect

currentdir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
parentdir = os.path.dirname(os.path.dirname(currentdir))
sys.path.insert(0,parentdir)

from utils.args import args

import setup.categories.ae_setup as AESetup
from models.autoencoders import *
from datasets.PCAM import PCAM

if __name__ == "__main__":
    dataset = PCAM(root_path=os.path.join(args.root_path, "pcam"), extract=True, downsample=64)
    model = ALILikeVAE(dims=(3, 64, 64))
    AESetup.train_variational_autoencoder(args, model=model, dataset=dataset.get_D1_train(), BCE_Loss=False)

示例#2

from __future__ import print_function
import os,sys,inspect

currentdir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
parentdir = os.path.dirname(os.path.dirname(currentdir))
sys.path.insert(0,parentdir)

from utils.args import args

import setup.categories.ae_setup as AESetup
from models.autoencoders import *
from datasets.PCAM import PCAM

if __name__ == "__main__":
    dataset = PCAM(root_path=os.path.join(args.root_path, "pcam"), extract=True, downsample=64)
    model = Generic_AE(dims=(3, 64, 64), max_channels=512, depth=12, n_hidden=512)
    AESetup.train_autoencoder(args, model=model, dataset=dataset.get_D1_train(), BCE_Loss=True)

示例#3

文件： PCAMTrainALILikeVAEBCE.py 项目： caotians1/OD-test-master

from __future__ import print_function
import os, sys, inspect

currentdir = os.path.dirname(
    os.path.abspath(inspect.getfile(inspect.currentframe())))
parentdir = os.path.dirname(os.path.dirname(currentdir))
sys.path.insert(0, parentdir)

from utils.args import args

import setup.categories.ae_setup as AESetup
from models.autoencoders import *
from datasets.PCAM import PCAM

if __name__ == "__main__":
    dataset = PCAM(root_path=os.path.join(args.root_path, "pcam"),
                   extract=True,
                   downsample=64)
    model = ALILikeVAE(dims=(3, 64, 64))
    AESetup.train_variational_autoencoder(args,
                                          model=model,
                                          dataset=dataset.get_D1_train(),
                                          BCE_Loss=True)

示例#4

文件： PCAMTrain.py 项目： caotians1/OD-test-master

from __future__ import print_function
import os,sys,inspect

currentdir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
parentdir = os.path.dirname(os.path.dirname(currentdir))
sys.path.insert(0,parentdir)

from utils.args import args
import setup.categories.classifier_setup as CLSetup
from models.classifiers import PCAMDense
from datasets.PCAM import PCAM


if __name__ == "__main__":
    dataset = PCAM(root_path=os.path.join(args.root_path, "pcam"), extract=True, downsample=224)
    model = PCAMDense("densenet121-a639ec97.pth", train_features=True)
    CLSetup.train_classifier(args, model=model, dataset=dataset.get_D1_train(), balanced=True)