def compute_features()

in rmac_features.py [0:0]

• 25 lines of code
• 1 McCabe index (conditional complexity)

def compute_features(cnn, args):
    pca = PCA(args.pca_dimensions, args.device)
    pca.DVt = torch.load(
        os.path.join(
            args.dataset_folder,
            args.pca_files_prefix + "_Dvt_resnet34_%d.t7" % args.resnet_level,
        ),
        map_location="cpu",
    ).to(args.device)
    pca.mean = torch.load(
        os.path.join(
            args.dataset_folder,
            args.pca_files_prefix + "_mean_resnet34_%d.t7" % args.resnet_level,
        ),
        map_location="cpu",
    ).to(args.device)

    in_filename = args.dataset_folder
    print("Computing descriptors for %s" % in_filename)
    descriptors = get_rmac_descriptors(cnn, args, in_filename, pca=pca)

    filename = os.path.basename(in_filename)
    filename, _ = os.path.splitext(filename)
    out_folder = os.path.join(args.out_folder, f"rmac_resnet34_{args.resnet_level}")
    fname_desc = os.path.join(out_folder, filename + ".pkl")
    os.makedirs(out_folder, exist_ok=True)
    pkl.dump(descriptors, open(fname_desc, "wb"), -1)