added yolov4 on toys lab, added both hands and reach the closest object

igor-lirussi · Sep 29, 2023 · 39a70f4 · 39a70f4
1 parent c400e0e
commit 39a70f4
Show file tree

Hide file tree

Showing 8 changed files with 1,283 additions and 40 deletions.
diff --git a/.gitignore b/.gitignore
@@ -12,6 +12,7 @@ models/yolov4.weights
 models/yolov4_new.weights
 models/yolov4x-mish.weights
 models/yolov4-csp.weights
+models/yolov4-colors.weights
 
 
 

diff --git a/README.md b/README.md
@@ -1,21 +1,41 @@
 # Baxter-Robot-ObjDet
 Repository for some experiments with Baxter robot and Object Detection.
 
-### Jupyter Notebook
-The notebook contains some examples of image processing with OpenCV and Pyhton.
-If it's not in the right path, open the notebook with Anaconda Propt passing the correct disk.
-Example:
+The code allows the robot:
+- to detect different objects,
+- approach them with his arm, indipendently from their position on the table in front of it,
+- pick them up safely and move them.
+
+> The models of large size in the folder models can be downloaded from the [releases](https://github.com/igor-lirussi/Baxter-Robot-ObjDet/releases)
+
+### Jupyter Notebooks
+The notebooks contain some examples of image processing with OpenCV and Pyhton.
+
+The notebooks study both face detection and object-detection.
+
+If it's not in the right path, open the notebook with Anaconda Propt passing the correct disk. Example:
 '''
 jupyter notebook --notebook-dir=D:
 '''
 
+### Colors Toys Object Detection Model
+A Yolo v4 model has been trained to detect 9 different classes of food toys present in CoLoRs Lab. 
+
+In the [releases](https://github.com/igor-lirussi/Baxter-Robot-ObjDet/releases) you can download:
+- colors.names
+- yolov4-colors.cfg
+- yolov4-colors.weights
+
+!(dataset)[img/dataset.jpg]
+
 ## Built With
 
 * Python 3.7.18
 
 ## Changelog
 
 **Version 1.0** - Initial release, 31-01-2022
+**Version 1.1** - Added Colors yolov4 trained net, 28-09-2023
 
 ## Authors
 

diff --git a/baxter-python3 b/baxter-python3
diff --git a/img/dataset.jpg b/img/dataset.jpg
diff --git a/models/-README.md b/models/-README.md
@@ -0,0 +1,2 @@
+## Model folder
+Large files of this folder (es weights of the neural networks) can be downloaded manually from the [releases](https://github.com/igor-lirussi/Baxter-Robot-ObjDet/releases)
diff --git a/models/colors.names b/models/colors.names
@@ -0,0 +1,9 @@
+apple
+cabbage
+carrot
+cheese
+grapes
+lemon
+mushroom
+strawberry
+tomato
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		## Model folder
		Large files of this folder (es weights of the neural networks) can be downloaded manually from the [releases](https://github.com/igor-lirussi/Baxter-Robot-ObjDet/releases)