deep_dream_model.cpp File Reference

#include "deep_dream_model.hpp"
#include "input_validation.hpp"
#include <torch/cuda.h>
#include <torch/nn/functional/pooling.h>
#include <torch/nn/functional/upsampling.h>
#include <torch/script.h>
#include <torch/torch.h>
#include <cuda_runtime_api.h>
#include <opencv2/cudaarithm.hpp>
#include <opencv2/cudawarping.hpp>
#include <opencv2/imgproc.hpp>
#include <algorithm>
#include <charconv>
#include <cmath>
#include <filesystem>
#include <limits>
#include <ostream>
#include <stdexcept>
#include <string>
#include <unordered_set>
#include <utility>

Go to the source code of this file.

Classes
struct	acmxvk::dream::Model::Impl
struct	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::TensorAscentResult

Namespaces
namespace	acmxvk
namespace	acmxvk::dream
namespace	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}

Functions
void	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::check_cuda (cudaError_t result, std::string_view operation)
std::vector< double >	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::double_list_attribute (const torch::jit::Module &module, const std::string &name)
std::vector< torch::Tensor >	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::feature_outputs (const c10::IValue &value)
std::int64_t	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::integer_attribute (const torch::jit::Module &module, const std::string &name)
std::vector< std::int64_t >	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::integer_list_attribute (const torch::jit::Module &module, const std::string &name)
torch::Tensor	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::normalization_tensor (const std::vector< double > &values, const torch::Device &device, torch::ScalarType scalar_type)
ModelMetadata	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::read_metadata (const torch::jit::Module &module)
c10::IValue	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::require_attribute (const torch::jit::Module &module, const std::string &name)
std::size_t	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::resolve_layer (const ModelMetadata &metadata, std::string_view selector)
std::string	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::string_attribute (const torch::jit::Module &module, const std::string &name)
std::vector< std::string >	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::string_list_attribute (const torch::jit::Module &module, const std::string &name)
void	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::validate_gradient_options (const GradientAscentOptions &options)
void	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::validate_normalization (const std::vector< double > &values, bool standard_deviation)

Variables
constexpr float	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::GRADIENT_EPSILON = 1.0e-8F
constexpr int	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::MAX_DREAM_DIMENSION = 4096
constexpr float	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::MAX_FEEDBACK = 0.99F
constexpr int	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::MAX_GRADIENT_ASCENT_ITERATIONS = 100
constexpr float	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::MAX_GRADIENT_ASCENT_STEP = 10.0F
constexpr int	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::MAX_JITTER = 64
constexpr std::size_t	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::MAX_LAYERS = 128
constexpr std::uintmax_t	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::MAX_MODEL_BYTES = 2ULL * 1024ULL * 1024ULL * 1024ULL
constexpr float	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::MAX_OCTAVE_SCALE = 3.0F
constexpr int	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::MAX_OCTAVES = 8
constexpr float	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::MAX_ROTATION_DEGREES = 5.0F
constexpr int	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::MAX_SMOOTHING = 16
constexpr std::int64_t	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::MAX_SOURCE_LAYER_INDEX = 4096
constexpr int	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::MAX_TARGET_CHANNEL = 65535
constexpr std::int64_t	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::MAX_TEST_INPUT_SIZE = 256
constexpr float	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::MAX_ZOOM = 1.1F
constexpr float	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::MIN_OCTAVE_SCALE = 1.1F
constexpr float	acmxvk::dream::anonymous_namespace{deep_dream_model.cpp}::MIN_ZOOM = 0.9F