pranav-bot

import tensorflow as tf
import numpy as np
import os
import matplotlib.pyplot as plt

# Paths to input images, heatmaps, and keypoints directories
input_image_dir = 'iMAgE/iMAgE/'
heatmap_dir = 'heATMAP/heATMAP/'
keypoints_dir = 'kpts2D/kpts2D/'

pranav-bot

import os
import torch
from torch.utils.data import Dataset, DataLoader
import pandas as pd
import numpy as np
from PIL import Image

class DroneDataset(Dataset):
    def __init__(self, datacdn_path, trajectories_path, intrinsic_matrix, keypoints_3d):
        """

pranav-bot

import os
import torch
from torch.utils.data import Dataset, DataLoader
import pandas as pd
import numpy as np
from PIL import Image


class DroneDataset(Dataset):
    def __init__(self, datacdn_path, trajectories_path, intrinsic_matrix):