It's the base code of my ShapeAnnotator's shapes-v1.pkl model.
You can download it.
Download python file (shape.py)
you may also copy this code.
import joblib
import numpy as np
import os
import cv2
import matplotlib.pyplot as plt
from sklearn.linear_model import LogisticRegression
classes = {"circle":1,
"rectangle":2,
"triangle":3 }
X=[]
Y=[]
path1="trainer dataset's folder"
main_folder=os.listdir(path1)
print(main_folder)
for folder in main_folder:… See the full description on the dataset page:
https://huggingface.co/datasets/Nayon01/trainer_files.
