Week 5
Sept 25th 2023
- Testing and documenting findings on the completed AI3 training job.
- Results from testing finetuned LLAMA2:
- Got good results on EQ data, but not for fire data and weather data. The input and output samples are documented here: https://projects.gqc.com/internal/Hackathon/AI3/llama#results
Sept 26th 2023
- Document AI3 Phase 3 progress as a slide deck (Sent in the mail and located on local in phase_3 folder)
- Update documentation on AI3 as part of the process.
- Conversation with SK and PS about AI3 summary PPT - Multi-model approach usage has to be mentioned and explained.
- Notes for meeting on Thursday with AI3 team
- Start with outstanding questions (Go through them once again)
- Mention that we are using LLMs for common sense understanding of the data and that it is working well.
- We need to know how they are going to measure the accuracy using NK keys
- Is there a mechanism of using unique identifier for JSON objects, that will help in picking the objects with NK keys from entire JSON message and comparing that object (with actual value before NK) with predicted object (with NK replaced by a standard key)?
- Or are they just going to compare and find the difference between the actual JSON and Predicted JSON as a whole, without comparing each object? This means if there is any difference, then our prediction of NK keys is wrong.
- Feedback on our report
Sept 27th 2023
- Update
AI3-Hackathon-SummaryPPT with comments from yesterday. - Sent the
Progress-Summary-and-QuestionsPPT slide with questions and updates in mail. - YOLO - Looking into improving the results
- Need to label the images to train it on using tools like
labelme. - [SK] Ask if we want to train YOLO using disaster city videos by extracting frames using
ffmpegand then label all those frames.- Update - Sudhir said to roll forward with this. Will be updating if we find anything else.
- Need to label the images to train it on using tools like
Sept 28th 2023
- Meeting with Jian Tao - Asked the questions from
Progress-Summary-and-QuestionsPPT and documented all the notes and answers on projects sites under bothQuestionsandTODOpages. - Explored on tools that can be used for YOLO training data.
- Discussed with Sudhir about the meeting and documented more questions to ask and some more notes.
Sept 29th 2023
YOLO finetuning research
List of resources found:
- https://medium.com/@yongsun.yoon/fine-tuning-yolov8-using-custom-dataset-generated-by-open-world-object-detector-5724e267645d
- https://learnopencv.com/train-yolov8-on-custom-dataset/
- https://encord.com/blog/automate-video-annotation-guide/#:~:text=In%20automated%20video%20annotation%20or,and%20end%20of%20a%20video.
- https://youtu.be/DMRlOWfRBKU?si=OKf95AbxtJ9Y5Aq3
Approach to fine-tuning:
Download videos from the GDrive for PSIAP
Extract frames from the videos using
ffmpegAnnotate the frames with objects of intrest using any tool
- Manually using tools like
LabelMe - Automated using tools like
Grounding DINO
- Manually using tools like
Save the annotations in YOLO supported formats, like JSON files for each frame.
Prepare a YAML file with paths to above data and parameters like epochs etc.
Run the command for training YOLO and provide
data=abc.yamlin the command. Example is below:yolo task=detect mode=train model=...pt data=abc.yaml epochs=50 imgsz=640The trained model is saved in the
runs/detect/train/weightsfolder. We will use thebest.ptmodel.
Updating AI3 questions from yesterday discussion with Sudhir.
Get updates on all AI3 projects - Monday.