As stated i have succeeded in running the src/inference/inference_image.py by adding the import of opencv (import cv2), meanwhile the current repo code does not include it, also the os library is imported twice.
import os
import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
import torchvision
from torchvision import datasets,transforms,models,utils
import numpy as np
import matplotlib.pyplot as plt
import os
import pandas as pd
from PIL import Image
import sys
import random
import shutil
from model import Detector
import argparse
from datetime import datetime
from tqdm import tqdm
from retinaface.pre_trained_models import get_model
from preprocess import extract_face
import warnings
warnings.filterwarnings('ignore')
Can be a quick fix by changing a few lines and pushing :smile:
As stated i have succeeded in running the src/inference/inference_image.py by adding the import of opencv (import cv2), meanwhile the current repo code does not include it, also the os library is imported twice.
Can be a quick fix by changing a few lines and pushing :smile: